Searched refs:GetInstructionWithName (Results 1 – 10 of 10) sorted by relevance
650 module->entry_computation()->GetInstructionWithName("broadcast.0"); in TEST_F()652 module->entry_computation()->GetInstructionWithName("broadcast.1"); in TEST_F()654 module->entry_computation()->GetInstructionWithName("reduce.1"); in TEST_F()656 module->entry_computation()->GetInstructionWithName("reduce.2"); in TEST_F()692 module->entry_computation()->GetInstructionWithName("broadcast.0"); in TEST_F()694 module->entry_computation()->GetInstructionWithName("reduce.1"); in TEST_F()733 module->entry_computation()->GetInstructionWithName("broadcast.0"); in TEST_F()737 module->entry_computation()->GetInstructionWithName("reduce.2"); in TEST_F()740 module->entry_computation()->GetInstructionWithName("reduce.3"); in TEST_F()
1125 module_->entry_computation()->GetInstructionWithName("fusion"); in TEST_F()1127 module_->entry_computation()->GetInstructionWithName("negate0"); in TEST_F()1129 module_->entry_computation()->GetInstructionWithName("negate1"); in TEST_F()1131 module_->entry_computation()->GetInstructionWithName("negate2"); in TEST_F()1170 module_->entry_computation()->GetInstructionWithName("fusion"); in TEST_F()1172 module_->entry_computation()->GetInstructionWithName("negate0"); in TEST_F()
1826 module->entry_computation()->GetInstructionWithName("while"); in TEST_P()2017 module->GetComputationWithName("entry")->GetInstructionWithName("copy"); in TEST_P()2020 ->GetInstructionWithName("neg1"); in TEST_P()2025 ->GetInstructionWithName("neg2"); in TEST_P()2080 module->GetComputationWithName("entry")->GetInstructionWithName( in TEST_P()2084 module->GetComputationWithName("entry")->GetInstructionWithName( in TEST_P()2088 ->GetInstructionWithName("add"); in TEST_P()2144 module->GetComputationWithName("entry")->GetInstructionWithName( in TEST_P()2148 ->GetInstructionWithName("add0"); in TEST_P()2153 module->GetComputationWithName("entry")->GetInstructionWithName("add1"); in TEST_P()[all …]
181 module_->entry_computation()->GetInstructionWithName("custom-call.1"); in TEST_F()183 module_->entry_computation()->GetInstructionWithName("custom-call.2"); in TEST_F()251 module_->entry_computation()->GetInstructionWithName("custom-call.1"); in TEST_F()
1928 m->entry_computation()->GetInstructionWithName("get-tuple-element.4"); in TEST_F()1930 m->entry_computation()->GetInstructionWithName("dynamic-update-slice.5"); in TEST_F()1932 m->entry_computation()->GetInstructionWithName("dynamic-update-slice.9"); in TEST_F()1972 m->entry_computation()->GetInstructionWithName("constant.1.1"); in TEST_F()1974 m->entry_computation()->GetInstructionWithName("constant.1.2"); in TEST_F()2239 m->entry_computation()->GetInstructionWithName("constant.42"); in TEST_F()2507 HloInstruction* custom_call = main->GetInstructionWithName("custom_call"); in TEST_F()
482 HloInstruction* GetInstructionWithName(absl::string_view name);
815 module->entry_computation()->GetInstructionWithName("infeed"); in TEST_F()817 module->entry_computation()->GetInstructionWithName("outfeed"); in TEST_F()
1146 HloInstruction* HloComputation::GetInstructionWithName(absl::string_view name) { in GetInstructionWithName() function in xla::HloComputation
1180 HloInstruction* dus = module->entry_computation()->GetInstructionWithName( in TEST_F()
1698 entry->GetInstructionWithName("t"); in TEST_P()1701 entry->GetInstructionWithName("t")); in TEST_P()