Searched refs:GetHloModule (Results 1 – 2 of 2) sorted by relevance
79 std::unique_ptr<HloModule> GetHloModule(const string& hlo_text) { in GetHloModule() function in xla::__anon655019a10111::DynamicPadderTest162 module_ = GetHloModule(hlo_text); in TEST_F()218 module_ = GetHloModule(hlo_text); in TEST_F()374 std::unique_ptr<HloModule> GetHloModule(const string& hlo_text) { in GetHloModule() function in xla::__anon655019a10111::ExecutionTest424 auto module_not_padded = GetHloModule(hlo_text_not_padded); in XLA_TEST_F()439 auto module_padded = GetHloModule(hlo_text_padded); in XLA_TEST_F()486 auto module_not_padded = GetHloModule(hlo_text); in XLA_TEST_F()496 auto module_padded = GetHloModule(hlo_text); in XLA_TEST_F()558 auto module = GetHloModule(hlo_text); in XLA_TEST_F()592 auto module_not_padded = GetHloModule(hlo_text_not_padded); in XLA_TEST_F()[all …]
83 StatusOr<std::shared_ptr<HloModule>> GetHloModule( in GetHloModule() function97 GetHloModule(computation)); in GetComputationHloText()108 GetHloModule(computation)); in GetComputationHloDotGraph()117 GetHloModule(computation)); in HashComputation()284 .def("get_hlo_module", &GetHloModule) in BuildXlaCompilerSubmodule()290 .def("as_hlo_module", &GetHloModule); in BuildXlaCompilerSubmodule()