@@ -122,7 +122,7 @@ TEST_F(ModuleTest, TestNonExistentMethodMeta) {
122122
123123TEST_F (ModuleTest, TestExecute) {
124124 Module module (model_path_);
125- auto tensor = make_tensor_ptr ({1 });
125+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
126126
127127 const auto result = module .execute (" forward" , {tensor, tensor});
128128 EXPECT_TRUE (result.ok ());
@@ -141,7 +141,7 @@ TEST_F(ModuleTest, TestExecutePreload) {
141141 const auto error = module .load ();
142142 EXPECT_EQ (error, Error::Ok);
143143
144- auto tensor = make_tensor_ptr ({1 });
144+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
145145
146146 const auto result = module .execute (" forward" , {tensor, tensor});
147147 EXPECT_TRUE (result.ok ());
@@ -157,7 +157,7 @@ TEST_F(ModuleTest, TestExecutePreload_method) {
157157 const auto error = module .load_method (" forward" );
158158 EXPECT_EQ (error, Error::Ok);
159159
160- auto tensor = make_tensor_ptr ({1 });
160+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
161161
162162 const auto result = module .execute (" forward" , {tensor, tensor});
163163 EXPECT_TRUE (result.ok ());
@@ -176,7 +176,7 @@ TEST_F(ModuleTest, TestExecutePreloadProgramAndMethod) {
176176 const auto load_method_error = module .load_method (" forward" );
177177 EXPECT_EQ (load_method_error, Error::Ok);
178178
179- auto tensor = make_tensor_ptr ({1 });
179+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
180180
181181 const auto result = module .execute (" forward" , {tensor, tensor});
182182 EXPECT_TRUE (result.ok ());
@@ -205,7 +205,7 @@ TEST_F(ModuleTest, TestExecuteOnCurrupted) {
205205TEST_F (ModuleTest, TestGet) {
206206 Module module (model_path_);
207207
208- auto tensor = make_tensor_ptr ({1 });
208+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
209209
210210 const auto result = module .get (" forward" , {tensor, tensor});
211211 EXPECT_TRUE (result.ok ());
@@ -280,7 +280,7 @@ TEST_F(ModuleTest, TestProgramSharingAndDataLoaderManagement) {
280280 EXPECT_EQ (load_error, Error::Ok);
281281 EXPECT_TRUE (module1->is_loaded ());
282282
283- auto tensor = make_tensor_ptr ({1 });
283+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
284284
285285 const auto result1 = module1->execute (" forward" , {tensor, tensor});
286286 EXPECT_TRUE (result1.ok ());
@@ -325,7 +325,7 @@ TEST_F(ModuleTest, TestProgramPersistenceAndReuseAfterModuleDestruction) {
325325
326326 EXPECT_EQ (module .program (), shared_program);
327327
328- auto tensor = make_tensor_ptr ({1 });
328+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
329329
330330 const auto result = module .execute (" forward" , {tensor, tensor});
331331 EXPECT_TRUE (result.ok ());
0 commit comments