Searched refs:FindComputation (Results 1 – 6 of 6) sorted by relevance
62 HloComputation* FindComputation(const HloComputation* old_computation) const { in FindComputation() function
595 if ((new_computation = context->FindComputation(computation)) != nullptr) { in DeepCloneComputation()
1569 context->FindComputation(fused_instructions_computation()); in CloneWithNewOperandsImpl()
131 HloComputation* FindComputation(const HloModule& module, in FindComputation() function253 const HloComputation* comp = FindComputation(module, node_name); in DoInfoCommand()
300 HloComputation* FindComputation(HloModule* module, absl::string_view name);
497 HloComputation* HloTestBase::FindComputation(HloModule* module, in FindComputation() function in xla::HloTestBase