Searched refs:ExtractModule (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | hlo_extractor_test.cc | 46 ExtractModule(FindInstruction(hlo_module.get(), "exp")); in TEST_F()
53 ExtractModule(FindInstruction(hlo_module.get(), "exp"), /*height=*/0); in TEST_F()
59 auto extracted_module = ExtractModule( in TEST_F()
84 ExtractModule(FindInstruction(hlo_module.get(), "exp")); in TEST_F()
91 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/0); in TEST_F()
97 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/1); in TEST_F()
104 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/2); in TEST_F()
128 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/0); in TEST_F()
134 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/1); in TEST_F()
|
| H A D | hlo_extractor.h | 31 std::unique_ptr<HloModule> ExtractModule(HloInstruction* instruction,
|
| H A D | hlo_extractor.cc | 138 std::unique_ptr<HloModule> ExtractModule(HloInstruction* instruction, in ExtractModule() function
|
| H A D | interactive_graphviz.cc | 402 auto extracted_module = ExtractModule(instr, height); in DoExtractCommand()
|