Home
last modified time | relevance | path

Searched defs:HandleCrossReplicaSum (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization.cc147 Status BFloat16NormalizationVisitor::HandleCrossReplicaSum( in HandleCrossReplicaSum() function in xla::BFloat16NormalizationVisitor
Ddfs_hlo_visitor_with_default.h91 Status HandleCrossReplicaSum(HloInstructionPtr crs) override { in HandleCrossReplicaSum() function
Dhlo_verifier.cc92 Status ShapeVerifier::HandleCrossReplicaSum(HloInstruction* crs) { in HandleCrossReplicaSum() function in xla::ShapeVerifier
Dhlo_cost_analysis.cc414 Status HloCostAnalysis::HandleCrossReplicaSum(const HloInstruction* crs) { in HandleCrossReplicaSum() function in xla::HloCostAnalysis
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc618 Status IrEmitter::HandleCrossReplicaSum(HloInstruction* crs) { in HandleCrossReplicaSum() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1193 Status IrEmitter::HandleCrossReplicaSum(HloInstruction* crs) { in HandleCrossReplicaSum() function in xla::cpu::IrEmitter