Home
last modified time | relevance | path

Searched defs:HandleAllReduce (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_conversion_folding.cc190 Status BFloat16ConversionFoldingVisitor::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::BFloat16ConversionFoldingVisitor
Ddfs_hlo_visitor_with_default.h104 Status HandleAllReduce(HloInstructionPtr crs) override { in HandleAllReduce() function
Dhlo_verifier.cc280 Status ShapeVerifier::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::ShapeVerifier
1877 Status HandleAllReduce(HloInstruction* crs) override { in HandleAllReduce() function in xla::__anon45533f2c0811::InstructionVerifier
Dhlo_cost_analysis.cc748 Status HloCostAnalysis::HandleAllReduce(const HloInstruction* crs) { in HandleAllReduce() function in xla::HloCostAnalysis
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc567 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc3178 Status IrEmitterUnnested::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1201 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2230 Status SpmdPartitioningVisitor::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::spmd::SpmdPartitioningVisitor