@@ -122,7 +122,7 @@ TEST_F(ModuleTest, TestNonExistentMethodMeta) {
122
122
123
123
TEST_F (ModuleTest, TestExecute) {
124
124
Module module (model_path_);
125
- auto tensor = make_tensor_ptr ({1 });
125
+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
126
126
127
127
const auto result = module .execute (" forward" , {tensor, tensor});
128
128
EXPECT_TRUE (result.ok ());
@@ -141,7 +141,7 @@ TEST_F(ModuleTest, TestExecutePreload) {
141
141
const auto error = module .load ();
142
142
EXPECT_EQ (error, Error::Ok);
143
143
144
- auto tensor = make_tensor_ptr ({1 });
144
+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
145
145
146
146
const auto result = module .execute (" forward" , {tensor, tensor});
147
147
EXPECT_TRUE (result.ok ());
@@ -157,7 +157,7 @@ TEST_F(ModuleTest, TestExecutePreload_method) {
157
157
const auto error = module .load_method (" forward" );
158
158
EXPECT_EQ (error, Error::Ok);
159
159
160
- auto tensor = make_tensor_ptr ({1 });
160
+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
161
161
162
162
const auto result = module .execute (" forward" , {tensor, tensor});
163
163
EXPECT_TRUE (result.ok ());
@@ -176,7 +176,7 @@ TEST_F(ModuleTest, TestExecutePreloadProgramAndMethod) {
176
176
const auto load_method_error = module .load_method (" forward" );
177
177
EXPECT_EQ (load_method_error, Error::Ok);
178
178
179
- auto tensor = make_tensor_ptr ({1 });
179
+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
180
180
181
181
const auto result = module .execute (" forward" , {tensor, tensor});
182
182
EXPECT_TRUE (result.ok ());
@@ -205,7 +205,7 @@ TEST_F(ModuleTest, TestExecuteOnCurrupted) {
205
205
TEST_F (ModuleTest, TestGet) {
206
206
Module module (model_path_);
207
207
208
- auto tensor = make_tensor_ptr ({1 });
208
+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
209
209
210
210
const auto result = module .get (" forward" , {tensor, tensor});
211
211
EXPECT_TRUE (result.ok ());
@@ -280,7 +280,7 @@ TEST_F(ModuleTest, TestProgramSharingAndDataLoaderManagement) {
280
280
EXPECT_EQ (load_error, Error::Ok);
281
281
EXPECT_TRUE (module1->is_loaded ());
282
282
283
- auto tensor = make_tensor_ptr ({1 });
283
+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
284
284
285
285
const auto result1 = module1->execute (" forward" , {tensor, tensor});
286
286
EXPECT_TRUE (result1.ok ());
@@ -325,7 +325,7 @@ TEST_F(ModuleTest, TestProgramPersistenceAndReuseAfterModuleDestruction) {
325
325
326
326
EXPECT_EQ (module .program (), shared_program);
327
327
328
- auto tensor = make_tensor_ptr ({1 });
328
+ auto tensor = make_tensor_ptr ({1 }, { 1 } );
329
329
330
330
const auto result = module .execute (" forward" , {tensor, tensor});
331
331
EXPECT_TRUE (result.ok ());
0 commit comments