Searched refs:HandleReplicaId (Results 1 – 7 of 7) sorted by relevance
79 Status HandleReplicaId(const HloInstruction* hlo) override;
109 Status HandleReplicaId(HloInstructionPtr hlo) override { in HandleReplicaId() function
60 Status HandleReplicaId(HloInstruction* hlo) override;
119 virtual Status HandleReplicaId(HloInstructionPtr hlo) = 0;
601 Status HloCostAnalysis::HandleReplicaId(const HloInstruction* /*hlo*/) { in HandleReplicaId() function in xla::HloCostAnalysis
204 Status ShapeVerifier::HandleReplicaId(HloInstruction* hlo) { in HandleReplicaId() function in xla::ShapeVerifier
2556 return visitor->HandleReplicaId(this); in Visit()