Searched defs:HandleAllReduceStart (Results 1 – 3 of 3) sorted by relevance
114 Status HandleAllReduceStart(HloInstructionPtr hlo) override { in HandleAllReduceStart() function
812 Status HloCostAnalysis::HandleAllReduceStart(const HloInstruction* hlo) { in HandleAllReduceStart() function in xla::HloCostAnalysis
465 Status ShapeVerifier::HandleAllReduceStart(HloInstruction* hlo) { in HandleAllReduceStart() function in xla::ShapeVerifier