Searched refs:CreateModuleWithProgramShape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_creation_utils_test.cc | 32 std::unique_ptr<VerifiedHloModule> CreateModuleWithProgramShape( in CreateModuleWithProgramShape() function in xla::__anonf5b064030111::HloCreationUtilsTest
47 std::unique_ptr<VerifiedHloModule> CreateModuleWithProgramShape( in CreateModuleWithProgramShape() function in xla::__anonf5b064030111::HloCreationUtilsTest
67 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{2}, in TEST_F()
86 auto module = CreateModuleWithProgramShape( in TEST_F()
109 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{2}, in TEST_F()
128 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{2}, in TEST_F()
147 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{}, in TEST_F()
166 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{6}, in TEST_F()
187 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{2}, in TEST_F()
207 auto module = CreateModuleWithProgramShape(S32, /*input_shape_dims=*/{}, in TEST_F()
[all …]
|