Home
last modified time | relevance | path

Searched refs:GetInstructionWithName (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test.cc650 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()
Dhlo_alias_analysis_test.cc1125 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()
Dmemory_space_assignment_test.cc1826 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 …]
Ddynamic_padder_test.cc181 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()
Dbuffer_assignment_test.cc1928 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()
Dhlo_computation.h482 HloInstruction* GetInstructionWithName(absl::string_view name);
Dhlo_cost_analysis_test.cc815 module->entry_computation()->GetInstructionWithName("infeed"); in TEST_F()
817 module->entry_computation()->GetInstructionWithName("outfeed"); in TEST_F()
Dhlo_computation.cc1146 HloInstruction* HloComputation::GetInstructionWithName(absl::string_view name) { in GetInstructionWithName() function in xla::HloComputation
Dbfloat16_propagation_test.cc1180 HloInstruction* dus = module->entry_computation()->GetInstructionWithName( in TEST_F()
Dhlo_dataflow_analysis_test.cc1698 entry->GetInstructionWithName("t"); in TEST_P()
1701 entry->GetInstructionWithName("t")); in TEST_P()