Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter.cc584 Status IrEmitter::HandleInfeed(HloInstruction*) { in HandleInfeed() function in xla::gpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddfs_hlo_visitor_with_default.h147 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed() function
H A Dhlo_cost_analysis.cc339 Status HloCostAnalysis::HandleInfeed(const HloInstruction* infeed) { in HandleInfeed() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc771 Status ShapeVerifier::HandleInfeed(HloInstruction* instruction) { in HandleInfeed() function in xla::ShapeVerifier
H A Dhlo_evaluator.cc1115 Status HloEvaluator::HandleInfeed(HloInstruction* infeed) { in HandleInfeed() function in xla::HloEvaluator
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc402 Status IrEmitter::HandleInfeed(HloInstruction* instruction) { in HandleInfeed() function in xla::cpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc3246 Status SpmdPartitioningVisitor::HandleInfeed(HloInstruction* hlo) { in HandleInfeed() function in xla::spmd::SpmdPartitioningVisitor