Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h81 Status HandleAllToAll(const HloInstruction* hlo) override;
Ddfs_hlo_visitor_with_default.h107 Status HandleAllToAll(HloInstructionPtr hlo) override { in HandleAllToAll() function
Dhlo_verifier.h61 Status HandleAllToAll(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h121 virtual Status HandleAllToAll(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc765 Status HloCostAnalysis::HandleAllToAll(const HloInstruction* hlo) { in HandleAllToAll() function in xla::HloCostAnalysis
Dhlo_verifier.cc291 Status ShapeVerifier::HandleAllToAll(HloInstruction* hlo) { in HandleAllToAll() function in xla::ShapeVerifier
Dhlo_instruction.cc3135 return visitor->HandleAllToAll(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h132 Status HandleAllToAll(HloInstruction* instruction) override;
Dir_emitter.cc1208 Status IrEmitter::HandleAllToAll(HloInstruction* instruction) { in HandleAllToAll() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h212 Status HandleAllToAll(HloInstruction* hlo) override;
Dir_emitter_unnested.cc3184 Status IrEmitterUnnested::HandleAllToAll(HloInstruction* hlo) { in HandleAllToAll() function in xla::gpu::IrEmitterUnnested