Home
last modified time | relevance | path

Searched refs:HandleOutfeed (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h85 Status HandleOutfeed(HloInstruction* outfeed) override;
Dir_emitter_unnested.h197 Status HandleOutfeed(HloInstruction* outfeed) override;
Dir_emitter.cc607 Status IrEmitter::HandleOutfeed(HloInstruction*) { in HandleOutfeed() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc3209 Status IrEmitterUnnested::HandleOutfeed(HloInstruction* outfeed) { in HandleOutfeed() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h286 Status HandleOutfeed(HloInstruction* outfeed) override { in HandleOutfeed() function
Dhlo_cost_analysis.h88 Status HandleOutfeed(const HloInstruction* outfeed) override;
Ddfs_hlo_visitor_with_default.h137 Status HandleOutfeed(HloInstructionPtr outfeed) override { in HandleOutfeed() function
Dhlo_verifier.h69 Status HandleOutfeed(HloInstruction*) override;
Ddfs_hlo_visitor.h235 virtual Status HandleOutfeed(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc352 Status HloCostAnalysis::HandleOutfeed(const HloInstruction* outfeed) { in HandleOutfeed() function in xla::HloCostAnalysis
Dhlo_verifier.cc431 Status ShapeVerifier::HandleOutfeed(HloInstruction* instruction) { in HandleOutfeed() function in xla::ShapeVerifier
Dhlo_instruction.cc3223 return visitor->HandleOutfeed(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h145 Status HandleOutfeed(HloInstruction* outfeed) override;
Dir_emitter.cc515 Status IrEmitter::HandleOutfeed(HloInstruction* outfeed) { in HandleOutfeed() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h420 Status HandleOutfeed(HloInstruction* hlo) override;
Dspmd_partitioner.cc2898 Status SpmdPartitioningVisitor::HandleOutfeed(HloInstruction* hlo) { in HandleOutfeed() function in xla::spmd::SpmdPartitioningVisitor