Home
last modified time | relevance | path

Searched refs:ParseAndReturnUnverifiedModule (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_verifier_test.cc176 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
197 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseAndReturnUnverifiedModule(hlo)); in TEST_F()
228 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
263 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
309 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
344 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
360 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
379 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
399 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
437 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
[all …]
H A Dmemory_space_propagation_test.cc124 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
193 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
267 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
341 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
404 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
467 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
H A Dall_gather_decomposer_test.cc49 ParseAndReturnUnverifiedModule((module_str))); in TEST_F()
72 ParseAndReturnUnverifiedModule((module_str))); in TEST_F()
95 ParseAndReturnUnverifiedModule((module_str))); in TEST_F()
118 ParseAndReturnUnverifiedModule((module_str))); in TEST_F()
144 ParseAndReturnUnverifiedModule((module_str))); in TEST_F()
H A Dhlo_parser.h39 StatusOr<std::unique_ptr<HloModule>> ParseAndReturnUnverifiedModule(
46 StatusOr<std::unique_ptr<HloModule>> ParseAndReturnUnverifiedModule(
H A Dhlo_runner_interface.cc27 return ParseAndReturnUnverifiedModule(hlo_string, config); in CreateModuleFromString()
71 return ParseAndReturnUnverifiedModule(hlo_string, config); in ReadModuleFromHloTextFile()
H A Ddynamic_padder_test.cc2086 auto module = ParseAndReturnUnverifiedModule(R"( in TEST_F()
2098 auto module = ParseAndReturnUnverifiedModule(R"( in TEST_F()
2115 auto module = ParseAndReturnUnverifiedModule(R"( in TEST_F()
2136 auto module = ParseAndReturnUnverifiedModule(R"( in TEST_F()
H A Dcopy_insertion_test.cc3282 ParseAndReturnUnverifiedModule(kModuleString)); in TEST_F()
3305 ParseAndReturnUnverifiedModule(kModuleString)); in TEST_F()
3327 ParseAndReturnUnverifiedModule(kModuleString)); in TEST_F()
3359 ParseAndReturnUnverifiedModule(kModuleString)); in TEST_F()
3379 ParseAndReturnUnverifiedModule(kModuleString)); in TEST_F()
H A Dhlo_parser_test.cc2797 ParseAndReturnUnverifiedModule(original).status().error_message(), in TEST_F()
2815 ParseAndReturnUnverifiedModule(original).status().error_message(), in TEST_F()
2843 ParseAndReturnUnverifiedModule(original).status().error_message(), in TEST_F()
H A Ddynamic_dimension_inference_test.cc1338 TF_ASSERT_OK_AND_ASSIGN(module_, ParseAndReturnUnverifiedModule(module_str)); in TEST_F()
H A Dhlo_parser.cc5933 StatusOr<std::unique_ptr<HloModule>> ParseAndReturnUnverifiedModule( in ParseAndReturnUnverifiedModule() function
5941 StatusOr<std::unique_ptr<HloModule>> ParseAndReturnUnverifiedModule( in ParseAndReturnUnverifiedModule() function
5943 return ParseAndReturnUnverifiedModule(str, HloModuleConfig()); in ParseAndReturnUnverifiedModule()
H A Dbuffer_assignment_test.cc1624 ParseAndReturnUnverifiedModule(kModuleString)); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client_test.cc47 ParseAndReturnUnverifiedModule(kProgram, {})); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dxla_mlir_translate.cc81 auto hlo_module_error = ParseAndReturnUnverifiedModule(content); in HloTextToMlirHloTranslateFunction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Dhlo_module_loader.cc83 ParseAndReturnUnverifiedModule(hlo_string, config)); in LoadModuleFromData()
H A Dreplay_computation.cc416 ParseAndReturnUnverifiedModule(hlo_module_text, config); in ParseSingleHloFile()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Djax_to_tfl_flatbuffer.cc104 auto hlo_module_error = xla::ParseAndReturnUnverifiedModule(content); in HloTextToMlirHloTranslateFunction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dtuple_test.cc368 ParseAndReturnUnverifiedModule(testcase)); in XLA_TEST_F()
H A Ddot_operation_test.cc1617 ParseAndReturnUnverifiedModule(hlo_string)); in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dmhlo_to_lhlo_with_xla.cc1746 xla::ParseAndReturnUnverifiedModule( in HloTextToLhloTranslateFunction()