Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator.h359 Status HandleOutfeed(HloInstruction* outfeed) override { in HandleOutfeed() function
H A Ddfs_hlo_visitor_with_default.h150 Status HandleOutfeed(HloInstructionPtr outfeed) override { in HandleOutfeed() function
H A Dhlo_cost_analysis.cc352 Status HloCostAnalysis::HandleOutfeed(const HloInstruction* outfeed) { in HandleOutfeed() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc781 Status ShapeVerifier::HandleOutfeed(HloInstruction* instruction) { in HandleOutfeed() function in xla::ShapeVerifier
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter.cc589 Status IrEmitter::HandleOutfeed(HloInstruction*) { in HandleOutfeed() function in xla::gpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc529 Status IrEmitter::HandleOutfeed(HloInstruction* outfeed) { in HandleOutfeed() function in xla::cpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc3679 Status SpmdPartitioningVisitor::HandleOutfeed(HloInstruction* hlo) { in HandleOutfeed() function in xla::spmd::SpmdPartitioningVisitor