Searched refs:HandleCollectivePermute (Results 1 – 7 of 7) sorted by relevance
78 Status HandleCollectivePermute(const HloInstruction* hlo) override;
106 Status HandleCollectivePermute(HloInstructionPtr hlo) override { in HandleCollectivePermute() function
59 Status HandleCollectivePermute(HloInstruction* hlo) override;
118 virtual Status HandleCollectivePermute(HloInstructionPtr hlo) = 0;
597 Status HloCostAnalysis::HandleCollectivePermute(const HloInstruction* /*hlo*/) { in HandleCollectivePermute() function in xla::HloCostAnalysis
208 Status ShapeVerifier::HandleCollectivePermute(HloInstruction* hlo) { in HandleCollectivePermute() function in xla::ShapeVerifier
2554 return visitor->HandleCollectivePermute(this); in Visit()