Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h82 Status HandleCollectivePermute(const HloInstruction* hlo) override;
Ddfs_hlo_visitor_with_default.h110 Status HandleCollectivePermute(HloInstructionPtr hlo) override { in HandleCollectivePermute() function
Dhlo_verifier.h62 Status HandleCollectivePermute(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h122 virtual Status HandleCollectivePermute(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc769 Status HloCostAnalysis::HandleCollectivePermute(const HloInstruction* /*hlo*/) { in HandleCollectivePermute() function in xla::HloCostAnalysis
Dhlo_verifier.cc369 Status ShapeVerifier::HandleCollectivePermute(HloInstruction* hlo) { in HandleCollectivePermute() function in xla::ShapeVerifier
Dhlo_instruction.cc3137 return visitor->HandleCollectivePermute(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h143 Status HandleCollectivePermute(HloInstruction* crs) override;
Dir_emitter.cc1263 Status IrEmitter::HandleCollectivePermute(HloInstruction* crs) { in HandleCollectivePermute() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h221 Status HandleCollectivePermute(HloInstruction* hlo) override;
Dir_emitter_unnested.cc3072 Status IrEmitterUnnested::HandleCollectivePermute(HloInstruction* hlo) { in HandleCollectivePermute() function in xla::gpu::IrEmitterUnnested