Searched refs:FindComputation (Results 1 – 6 of 6) sorted by relevance
62 HloComputation* FindComputation(const HloComputation* old_computation) const { in FindComputation() function
747 auto found_computation = context.FindComputation(computation); in RemoveUnusedComputations()762 if ((new_computation = context->FindComputation(computation)) != nullptr) { in DeepCloneComputation()
1832 context->FindComputation(fused_instructions_computation()); in CloneWithNewOperandsImpl()
136 HloComputation* FindComputation(const HloModule& module, in FindComputation() function279 const HloComputation* comp = FindComputation(module, node_name); in DoInfoCommand()
301 HloComputation* FindComputation(HloModule* module, absl::string_view name);
572 HloComputation* HloTestBase::FindComputation(HloModule* module, in FindComputation() function in xla::HloTestBase