Home
last modified time | relevance | path

Searched refs:GetModuleConfigForTest (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dall_reduce_test.cc47 ParseHloString(module_str, GetModuleConfigForTest()).ValueOrDie(); in XLA_TEST_F()
68 ParseHloString(module_str, GetModuleConfigForTest()).ValueOrDie(); in XLA_TEST_F()
94 ParseHloString(module_str, GetModuleConfigForTest()).ValueOrDie(); in XLA_TEST_F()
Dtoken_hlo_test.cc225 ParseHloString(module_string, GetModuleConfigForTest())); in XLA_TEST_F()
246 ParseHloString(module_string, GetModuleConfigForTest())); in XLA_TEST_F()
264 ParseHloString(module_string, GetModuleConfigForTest())); in XLA_TEST_F()
287 ParseHloString(module_string, GetModuleConfigForTest())); in XLA_TEST_F()
Dmulti_device_all_reduce_test.cc43 auto config = GetModuleConfigForTest(); in XLA_TEST_F()
Dhlo_test_base.cc135 return absl::make_unique<HloModule>(name, GetModuleConfigForTest()); in CreateNewUnverifiedModule()
141 name, GetModuleConfigForTest(), verifier_layout_sensitive_, in CreateNewVerifiedModule()
Dhlo_test_base.h158 HloModuleConfig GetModuleConfigForTest() { in GetModuleConfigForTest() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter_test.cc34 ->RunHloPasses(ParseHloString(hlo_string, GetModuleConfigForTest()) in GetOptimizedHlo()
297 ->RunHloPasses(ParseHloString(kHloString, GetModuleConfigForTest()) in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_kernel_tiling_test.cc35 return GetModuleConfigForTest(); in ConfigWithLayoutAssignment()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc1150 ParseAndReturnVerifiedModule(module_str, GetModuleConfigForTest())); in TEST_F()
1166 ParseAndReturnVerifiedModule(module_str, GetModuleConfigForTest())); in TEST_F()
1193 ParseAndReturnVerifiedModule(module_str, GetModuleConfigForTest())); in TEST_F()
1225 ParseAndReturnVerifiedModule(module_str, GetModuleConfigForTest())); in TEST_F()
1252 ParseAndReturnVerifiedModule(module_str, GetModuleConfigForTest())); in TEST_F()
1287 ParseAndReturnVerifiedModule(module_str, GetModuleConfigForTest())); in TEST_F()
Dhlo_dataflow_analysis_test.cc1920 ParseHloString(module_string, GetModuleConfigForTest())); in TEST_P()