Searched defs:HandleAllReduceDone (Results 1 – 3 of 3) sorted by relevance
117 Status HandleAllReduceDone(HloInstructionPtr hlo) override { in HandleAllReduceDone() function
816 Status HloCostAnalysis::HandleAllReduceDone(const HloInstruction* /*hlo*/) { in HandleAllReduceDone() function in xla::HloCostAnalysis
481 Status ShapeVerifier::HandleAllReduceDone(HloInstruction* hlo) { in HandleAllReduceDone() function in xla::ShapeVerifier