Searched refs:FindComputation (Results 1 – 6 of 6) sorted by relevance
62 HloComputation* FindComputation(const HloComputation* old_computation) const { in FindComputation() function
678 auto found_computation = context.FindComputation(computation); in RemoveUnusedComputations()693 if ((new_computation = context->FindComputation(computation)) != nullptr) { in DeepCloneComputation()
1687 context->FindComputation(fused_instructions_computation()); in CloneWithNewOperandsImpl()
135 HloComputation* FindComputation(const HloModule& module, in FindComputation() function259 const HloComputation* comp = FindComputation(module, node_name); in DoInfoCommand()
278 HloComputation* FindComputation(HloModule* module, absl::string_view name);
493 HloComputation* HloTestBase::FindComputation(HloModule* module, in FindComputation() function in xla::HloTestBase