Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h111 Status HandleSelectAndScatter(const HloInstruction* instruction) override;
Ddfs_hlo_visitor_with_default.h189 Status HandleSelectAndScatter(HloInstructionPtr select_and_scatter) override { in HandleSelectAndScatter() function
Dhlo_verifier.h94 Status HandleSelectAndScatter(HloInstruction* instruction) override;
Ddfs_hlo_visitor.h260 virtual Status HandleSelectAndScatter(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc127 Status HandleSelectAndScatter(HloInstruction* hlo) override;
1082 Status DynamicDimensionInferenceVisitor::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.cc438 Status HloCostAnalysis::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::HloCostAnalysis
Dhlo_verifier.cc910 Status ShapeVerifier::HandleSelectAndScatter(HloInstruction* instruction) { in HandleSelectAndScatter() function in xla::ShapeVerifier
Dhlo_evaluator_typed_visitor.h1840 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override {
Dhlo_instruction.cc3157 return visitor->HandleSelectAndScatter(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h150 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override;
Dir_emitter.cc664 Status IrEmitter::HandleSelectAndScatter(HloInstruction* select_and_scatter) { in HandleSelectAndScatter() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h428 Status HandleSelectAndScatter(HloInstruction* hlo) override;
Dspmd_partitioner.cc3147 Status SpmdPartitioningVisitor::HandleSelectAndScatter(HloInstruction* hlo) { in HandleSelectAndScatter() function in xla::spmd::SpmdPartitioningVisitor
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h192 Status HandleSelectAndScatter(HloInstruction* instruction) override;
Dir_emitter_unnested.cc2178 Status IrEmitterUnnested::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::gpu::IrEmitterUnnested