Searched refs:HandleReplicaId (Results 1 – 11 of 11) sorted by relevance
85 Status HandleReplicaId(const HloInstruction* hlo) override;
119 Status HandleReplicaId(HloInstructionPtr hlo) override { in HandleReplicaId() function
66 Status HandleReplicaId(HloInstruction* hlo) override;
125 virtual Status HandleReplicaId(HloInstructionPtr hlo) = 0;
787 Status HloCostAnalysis::HandleReplicaId(const HloInstruction* /*hlo*/) { in HandleReplicaId() function in xla::HloCostAnalysis
339 Status ShapeVerifier::HandleReplicaId(HloInstruction* hlo) { in HandleReplicaId() function in xla::ShapeVerifier
3143 return visitor->HandleReplicaId(this); in Visit()
170 Status HandleReplicaId(HloInstruction* hlo) override;
1301 Status IrEmitter::HandleReplicaId(HloInstruction* hlo) { in HandleReplicaId() function in xla::cpu::IrEmitter
218 Status HandleReplicaId(HloInstruction* hlo) override;
3060 Status IrEmitterUnnested::HandleReplicaId(HloInstruction* hlo) { in HandleReplicaId() function in xla::gpu::IrEmitterUnnested