Revert "Reshape only in row major layout (#687)"

This reverts commit 8152560.

runtime/lib/ttnn/program.cpp

@@ -710,14 +710,7 @@ vectorToArray(const std::vector<int32_t> &vec) {
 template <int32_t Rank>
 static ::ttnn::Tensor invoke_reshape(const ::ttnn::Tensor &tensor,
                                      const std::vector<int32_t> &shape) {
-  // TDOO #686 - figure out how to call reshape in tile layout
-  if (tensor.get_layout() == ::ttnn::Layout::ROW_MAJOR) {
-    return ::ttnn::reshape(tensor, vectorToArray<Rank>(shape));
-  }
-
-  auto rowMajorTensor = untilize(tensor);
-  auto res = ::ttnn::reshape(rowMajorTensor, vectorToArray<Rank>(shape));
-  return tilize(res);
+  return ::ttnn::reshape(tensor, vectorToArray<Rank>(shape));
 }
 
 static void run(::tt::target::ttnn::ReshapeOp const *op,

test/ttmlir/Dialect/TTNN/simple_reshape.mlir

@@ -0,0 +1,10 @@
+// RUN: ttmlir-opt --ttir-to-ttnn-backend-pipeline %s| FileCheck %s
+#any_device_tile = #tt.operand_constraint<dram|l1|tile|any_device_tile>
+module attributes {} {
+  func.func @forward(%arg0: tensor<4x2x32x32xbf16>) -> tensor<2x4x32x32xbf16> {
+    %0 = tensor.empty() : tensor<2x4x32x32xbf16>
+    // CHECK: %[[C:.*]] = "ttnn.reshape"[[C:.*]]
+    %1 = "ttir.reshape"(%arg0, %0) <{shape = [2: i32, 4: i32, 32: i32, 32: i32] , operand_constraints = [#any_device_tile, #any_device_tile]}> : (tensor<4x2x32x32xbf16>, tensor<2x4x32x32xbf16>) -> tensor<2x4x32x32xbf16>
+    return %1 : tensor<2x4x32x32xbf16>
+  }
+}