Searched refs:CreateModuleWithProgramShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils_test.cc | 30 std::unique_ptr<VerifiedHloModule> CreateModuleWithProgramShape( in CreateModuleWithProgramShape() function in xla::__anon5474724d0111::HloCreationUtilsTest
50 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{2}, in TEST_F()
69 auto module = CreateModuleWithProgramShape( in TEST_F()
92 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{2}, in TEST_F()
111 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{2}, in TEST_F()
130 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{}, in TEST_F()
149 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{6}, in TEST_F()
170 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{2}, in TEST_F()
190 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{}, in TEST_F()
209 auto module = CreateModuleWithProgramShape(F32, /*input_shape_dims=*/{}, in TEST_F()
|