Home
last modified time | relevance | path

Searched refs:HandleSelectAndScatter (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h73 Status HandleSelectAndScatter(HloInstruction* instruction) override;
Dir_emitter_unnested.cc1731 Status IrEmitterUnnested::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.h71 Status HandleSelectAndScatter(HloInstruction* instruction) override;
Dhlo_cost_analysis.h95 Status HandleSelectAndScatter(const HloInstruction* instruction) override;
Ddfs_hlo_visitor_with_default.h155 Status HandleSelectAndScatter(HloInstructionPtr select_and_scatter) override { in HandleSelectAndScatter() function
Ddfs_hlo_visitor.h214 virtual Status HandleSelectAndScatter(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc295 Status HloCostAnalysis::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::HloCostAnalysis
Dhlo_verifier.cc272 Status ShapeVerifier::HandleSelectAndScatter(HloInstruction* instruction) { in HandleSelectAndScatter() function in xla::ShapeVerifier
Dhlo_evaluator.cc1524 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override { in HandleSelectAndScatter() function in xla::HloEvaluator::TypedVisitor
Dhlo_instruction.cc2603 return visitor->HandleSelectAndScatter(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h135 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override;
Dir_emitter.cc614 Status IrEmitter::HandleSelectAndScatter(HloInstruction* select_and_scatter) { in HandleSelectAndScatter() function in xla::cpu::IrEmitter