| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_extractor_test.cc | 44 ExtractModule(FindInstruction(hlo_module.get(), "exp")); in TEST_F()
51 ExtractModule(FindInstruction(hlo_module.get(), "exp"), /*height=*/0); in TEST_F()
58 FindInstruction(hlo_module.get(), "negate"), /*height=*/0); in TEST_F()
82 ExtractModule(FindInstruction(hlo_module.get(), "exp")); in TEST_F()
89 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/0); in TEST_F()
95 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/1); in TEST_F()
102 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/2); in TEST_F()
126 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/0); in TEST_F()
132 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/1); in TEST_F()
|
| D | interactive_graphviz.cc | 113 HloInstruction* FindInstruction(const HloModule& module, string node_name) { in FindInstruction() function
252 const HloInstruction* instr = FindInstruction(module, node_name); in DoInfoCommand()
350 HloInstruction* instr = FindInstruction(module, node_name); in DoExtractCommand()
|
| D | hlo_extractor.cc | 88 clone_context_.FindInstruction(old_instruction)) { in FinishVisit()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_domain_test.cc | 50 HloInstruction* instruction = FindInstruction(module, instruction_name); in HasDomainEdge()
51 HloInstruction* operand = FindInstruction(module, operand_name); in HasDomainEdge()
281 HloInstruction* add = FindInstruction(module.get(), "c"); in TEST_F()
386 HloInstruction* infeed_data = FindInstruction(module.get(), "infeed.data"); in TEST_F()
459 HloInstruction* gte = FindInstruction(module.get(), "gte"); in TEST_F()
460 HloInstruction* tuple = FindInstruction(module.get(), "tuple"); in TEST_F()
519 auto tpl = FindInstruction(module.get(), "tpl"); in TEST_F()
617 auto tuple0 = FindInstruction(module.get(), "tuple.0"); in TEST_F()
|
| D | hlo_ordering_test.cc | 382 HloInstruction* add_1 = FindInstruction(module.get(), "add.1"); in TEST_F()
383 HloInstruction* add_2 = FindInstruction(module.get(), "add.2"); in TEST_F()
384 HloInstruction* add_3 = FindInstruction(module.get(), "add.3"); in TEST_F()
385 HloInstruction* conditional = FindInstruction(module.get(), "conditional"); in TEST_F()
|
| D | hlo_clone_context.h | 56 HloInstruction* FindInstruction(const HloInstruction* old_instruction) const { in FindInstruction() function
|
| D | layout_assignment_test.cc | 64 FindInstruction(m, name)->shape().layout().minor_to_major(); in LayoutOf()
740 EXPECT_THAT(FindInstruction(m.get(), "gte1") in TEST_F()
746 EXPECT_THAT(FindInstruction(m.get(), "gte1") in TEST_F()
878 ShapeUtil::GetSubshape(FindInstruction(m.get(), "send")->shape(), {0}), in TEST_F()
1299 const HloInstruction* custom_call = FindInstruction(m.get(), "custom-call"); in TEST_F()
|
| D | buffer_liveness_test.cc | 682 auto tuple_param0 = FindInstruction(module.get(), "param0"); in Run()
694 auto tuple_param0 = FindInstruction(module.get(), "param0"); in RunWithHloDataflowAnalysis()
|
| D | hlo_parser.cc | 107 std::pair<HloInstruction*, LocTy>* FindInstruction(
445 std::pair<HloInstruction*, HloParser::LocTy>* HloParser::FindInstruction( in FindInstruction() function in xla::__anon8590f6b60111::HloParser
1963 std::pair<HloInstruction*, LocTy>* instr = FindInstruction(name); in ParseInstructionNames()
2566 FindInstruction(name, shape); in ParseOperands()
|
| D | buffer_assignment_test.cc | 2556 auto dus9 = FindInstruction(module.get(), "dynamic-update-slice.9"); in TEST_F()
2559 auto dus5 = FindInstruction(module.get(), "dynamic-update-slice.5"); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_sanitize_constant_names_test.cc | 76 EXPECT_THAT(FindInstruction(module.get(), "equal_to_1"), op::Constant()); in TEST_F()
77 EXPECT_THAT(FindInstruction(module.get(), "equal_to_2"), op::Constant()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 301 HloInstruction* FindInstruction(HloModule* module, absl::string_view name);
|
| D | hlo_test_base.cc | 508 HloInstruction* HloTestBase::FindInstruction(HloModule* module, in FindInstruction() function in xla::HloTestBase
|