Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h283 Status HandleInfeed(HloInstruction* infeed) override { in HandleInfeed() function
Ddfs_hlo_visitor_with_default.h134 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed() function
Dhlo_cost_analysis.cc339 Status HloCostAnalysis::HandleInfeed(const HloInstruction* infeed) { in HandleInfeed() function in xla::HloCostAnalysis
Dhlo_verifier.cc421 Status ShapeVerifier::HandleInfeed(HloInstruction* instruction) { in HandleInfeed() function in xla::ShapeVerifier
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc602 Status IrEmitter::HandleInfeed(HloInstruction*) { in HandleInfeed() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc3190 Status IrEmitterUnnested::HandleInfeed(HloInstruction* xla_infeed) { in HandleInfeed() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc389 Status IrEmitter::HandleInfeed(HloInstruction* instruction) { in HandleInfeed() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2491 Status SpmdPartitioningVisitor::HandleInfeed(HloInstruction* hlo) { in HandleInfeed() function in xla::spmd::SpmdPartitioningVisitor