Update tensor_ptr_test.cpp

Author: shoumikhin

extension/tensor/test/tensor_ptr_test.cpp

@@ -24,7 +24,7 @@ class TensorPtrTest : public ::testing::Test {
 
 TEST_F(TensorPtrTest, ScalarTensorCreation) {
   float scalar_data = 3.14f;
-  auto tensor = make_tensor_ptr(exec_aten::ScalarType::Float, {}, &scalar_data);
+  auto tensor = make_tensor_ptr({}, &scalar_data);
 
   EXPECT_EQ(tensor->numel(), 1);
   EXPECT_EQ(tensor->dim(), 0);
@@ -80,8 +80,7 @@ TEST_F(TensorPtrTest, ScalarTensorSingleValueCreation) {
 
 TEST_F(TensorPtrTest, CreateTensorWithStridesAndDimOrder) {
   float data[20] = {2};
-  auto tensor = make_tensor_ptr(
-      exec_aten::ScalarType::Float, {4, 5}, data, {0, 1}, {5, 1});
+  auto tensor = make_tensor_ptr({4, 5}, data, {0, 1}, {5, 1});
   EXPECT_EQ(tensor->dim(), 2);
   EXPECT_EQ(tensor->size(0), 4);
   EXPECT_EQ(tensor->size(1), 5);
@@ -93,7 +92,7 @@ TEST_F(TensorPtrTest, CreateTensorWithStridesAndDimOrder) {
 
 TEST_F(TensorPtrTest, TensorSharingImpl) {
   float data[20] = {2};
-  auto tensor1 = make_tensor_ptr(exec_aten::ScalarType::Float, {4, 5}, data);
+  auto tensor1 = make_tensor_ptr({4, 5}, data);
   auto tensor2 = make_tensor_ptr(tensor1);
   EXPECT_EQ(tensor1->unsafeGetTensorImpl(), tensor2->unsafeGetTensorImpl());
 }
@@ -103,8 +102,7 @@ TEST_F(TensorPtrTest, TensorImplLifetime) {
   EXPECT_EQ(tensor, nullptr);
   {
     float data[20] = {2};
-    auto tensor_impl =
-        make_tensor_impl_ptr(exec_aten::ScalarType::Float, {4, 5}, data);
+    auto tensor_impl = make_tensor_impl_ptr({4, 5}, data);
     tensor = make_tensor_ptr(tensor_impl);
   }
   EXPECT_EQ(tensor->dim(), 2);
@@ -114,22 +112,22 @@ TEST_F(TensorPtrTest, TensorImplLifetime) {
 
 TEST_F(TensorPtrTest, TensorWithZeroDimensionAndElements) {
   float data[20] = {2};
-  auto tensor = make_tensor_ptr(exec_aten::ScalarType::Float, {}, data);
+  auto tensor = make_tensor_ptr({}, data);
   EXPECT_EQ(tensor->dim(), 0);
   EXPECT_EQ(tensor->numel(), 1);
-  tensor = make_tensor_ptr(exec_aten::ScalarType::Float, {0, 5}, data);
+  tensor = make_tensor_ptr({0, 5}, data);
   EXPECT_EQ(tensor->dim(), 2);
   EXPECT_EQ(tensor->numel(), 0);
 }
 
 TEST_F(TensorPtrTest, TensorResize) {
   float data[20] = {2};
   auto tensor = make_tensor_ptr(
-      exec_aten::ScalarType::Float,
       {4, 5},
       data,
       {},
       {},
+      exec_aten::ScalarType::Float,
       exec_aten::TensorShapeDynamism::DYNAMIC_UNBOUND);
   EXPECT_EQ(resize_tensor_ptr(tensor, {5, 4}), Error::Ok);
   EXPECT_EQ(tensor->size(0), 5);
@@ -138,7 +136,7 @@ TEST_F(TensorPtrTest, TensorResize) {
 
 TEST_F(TensorPtrTest, TensorDataAccess) {
   float data[6] = {1, 2, 3, 4, 5, 6};
-  auto tensor = make_tensor_ptr(exec_aten::ScalarType::Float, {2, 3}, data);
+  auto tensor = make_tensor_ptr({2, 3}, data);
   EXPECT_EQ(tensor->const_data_ptr<float>()[0], 1);
   EXPECT_EQ(tensor->const_data_ptr<float>()[5], 6);
   tensor->mutable_data_ptr<float>()[0] = 10;
@@ -149,11 +147,11 @@ TEST_F(TensorPtrTest, TensorWithCustomDataDeleter) {
   auto deleter_called = false;
   float* data = new float[20]();
   auto tensor = make_tensor_ptr(
-      exec_aten::ScalarType::Float,
       {4, 5},
       data,
       {},
       {},
+      exec_aten::ScalarType::Float,
       exec_aten::TensorShapeDynamism::DYNAMIC_BOUND,
       [&deleter_called](void* ptr) {
         deleter_called = true;
@@ -169,11 +167,11 @@ TEST_F(TensorPtrTest, TensorManagesMovedVector) {
   std::vector<float> data(20, 3.0f);
   auto* data_ptr = data.data();
   auto tensor = make_tensor_ptr(
-      exec_aten::ScalarType::Float,
       {4, 5},
       data_ptr,
       {},
       {},
+      exec_aten::ScalarType::Float,
       exec_aten::TensorShapeDynamism::DYNAMIC_BOUND,
       [moved_data = std::move(data), &deleter_called](void*) mutable {
         deleter_called = true;
@@ -191,11 +189,11 @@ TEST_F(TensorPtrTest, TensorDeleterReleasesCapturedSharedPtr) {
   std::shared_ptr<float[]> data_ptr(
       new float[10], [](float* ptr) { delete[] ptr; });
   auto tensor = make_tensor_ptr(
-      exec_aten::ScalarType::Float,
       {4, 5},
       data_ptr.get(),
       {},
       {},
+      exec_aten::ScalarType::Float,
       exec_aten::TensorShapeDynamism::DYNAMIC_BOUND,
       [data_ptr, &deleter_called](void*) mutable { deleter_called = true; });