Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h286 Status HandleOutfeed(HloInstruction* outfeed) override { in HandleOutfeed() function
Ddfs_hlo_visitor_with_default.h137 Status HandleOutfeed(HloInstructionPtr outfeed) override { in HandleOutfeed() function
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
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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/cpu/
Dir_emitter.cc515 Status IrEmitter::HandleOutfeed(HloInstruction* outfeed) { in HandleOutfeed() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2898 Status SpmdPartitioningVisitor::HandleOutfeed(HloInstruction* hlo) { in HandleOutfeed() function in xla::spmd::SpmdPartitioningVisitor