Home
last modified time | relevance | path

Searched refs:HandleReplicaId (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h85 Status HandleReplicaId(const HloInstruction* hlo) override;
Ddfs_hlo_visitor_with_default.h119 Status HandleReplicaId(HloInstructionPtr hlo) override { in HandleReplicaId() function
Dhlo_verifier.h66 Status HandleReplicaId(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h125 virtual Status HandleReplicaId(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc787 Status HloCostAnalysis::HandleReplicaId(const HloInstruction* /*hlo*/) { in HandleReplicaId() function in xla::HloCostAnalysis
Dhlo_verifier.cc339 Status ShapeVerifier::HandleReplicaId(HloInstruction* hlo) { in HandleReplicaId() function in xla::ShapeVerifier
Dhlo_instruction.cc3143 return visitor->HandleReplicaId(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h170 Status HandleReplicaId(HloInstruction* hlo) override;
Dir_emitter.cc1301 Status IrEmitter::HandleReplicaId(HloInstruction* hlo) { in HandleReplicaId() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h218 Status HandleReplicaId(HloInstruction* hlo) override;
Dir_emitter_unnested.cc3060 Status IrEmitterUnnested::HandleReplicaId(HloInstruction* hlo) { in HandleReplicaId() function in xla::gpu::IrEmitterUnnested