Home
last modified time | relevance | path

Searched refs:HandleAllGather (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h79 Status HandleAllGather(const HloInstruction* hlo) override;
Ddfs_hlo_visitor_with_default.h101 Status HandleAllGather(HloInstructionPtr crs) override { in HandleAllGather() function
Dhlo_verifier.h59 Status HandleAllGather(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h119 virtual Status HandleAllGather(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc744 Status HloCostAnalysis::HandleAllGather(const HloInstruction* hlo) { in HandleAllGather() function in xla::HloCostAnalysis
Dhlo_verifier.cc250 Status ShapeVerifier::HandleAllGather(HloInstruction* hlo) { in HandleAllGather() function in xla::ShapeVerifier
Dhlo_instruction.cc3131 return visitor->HandleAllGather(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h210 Status HandleAllGather(HloInstruction* hlo) override;
Dir_emitter_unnested.cc3172 Status IrEmitterUnnested::HandleAllGather(HloInstruction* hlo) { in HandleAllGather() function in xla::gpu::IrEmitterUnnested