Searched refs:GetInstructionWithName (Results 1 – 3 of 3) sorted by relevance
416 HloInstruction* GetInstructionWithName(absl::string_view name);
1823 m->entry_computation()->GetInstructionWithName("constant.1.1"); in TEST_F()1825 m->entry_computation()->GetInstructionWithName("constant.1.2"); in TEST_F()2091 m->entry_computation()->GetInstructionWithName("constant.42"); in TEST_F()2360 HloInstruction* custom_call = main->GetInstructionWithName("custom_call"); in TEST_F()
1044 HloInstruction* HloComputation::GetInstructionWithName(absl::string_view name) { in GetInstructionWithName() function in xla::HloComputation