Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc66 Status HandleSelectAndScatter(HloInstruction* hlo) override;
372 Status DynamicDimensionInferenceVisitor::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.h102 Status HandleSelectAndScatter(const HloInstruction* instruction) override;
Ddfs_hlo_visitor_with_default.h170 Status HandleSelectAndScatter(HloInstructionPtr select_and_scatter) override { in HandleSelectAndScatter() function
Dhlo_verifier.h85 Status HandleSelectAndScatter(HloInstruction* instruction) override;
Ddfs_hlo_visitor.h243 virtual Status HandleSelectAndScatter(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc328 Status HloCostAnalysis::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::HloCostAnalysis
Dhlo_verifier.cc636 Status ShapeVerifier::HandleSelectAndScatter(HloInstruction* instruction) { in HandleSelectAndScatter() function in xla::ShapeVerifier
Dhlo_evaluator_typed_visitor.h1856 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override { in HandleSelectAndScatter() function
Dhlo_instruction.cc2568 return visitor->HandleSelectAndScatter(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h170 Status HandleSelectAndScatter(HloInstruction* instruction) override;
Dir_emitter_unnested.cc781 Status IrEmitterUnnested::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h164 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override;
Dir_emitter.cc745 Status IrEmitter::HandleSelectAndScatter(HloInstruction* select_and_scatter) { in HandleSelectAndScatter() function in xla::cpu::IrEmitter