Home
last modified time | relevance | path

Searched defs:HandleReplicaId (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h119 Status HandleReplicaId(HloInstructionPtr hlo) override { in HandleReplicaId() function
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
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
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.cc3060 Status IrEmitterUnnested::HandleReplicaId(HloInstruction* hlo) { in HandleReplicaId() function in xla::gpu::IrEmitterUnnested