Home
last modified time | relevance | path

Searched refs:HandleScatter (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h88 Status HandleScatter(HloInstruction* scatter) override;
Dir_emitter.cc140 Status IrEmitter::HandleScatter(HloInstruction*) { in HandleScatter() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h167 Status HandleScatter(const HloInstruction* hlo) override;
Ddfs_hlo_visitor_with_default.h259 Status HandleScatter(HloInstructionPtr scatter) override { in HandleScatter() function
Dhlo_evaluator.h297 Status HandleScatter(HloInstruction* hlo) override;
Dalgebraic_simplifier.h335 Status HandleScatter(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h270 virtual Status HandleScatter(HloInstructionPtr hlo) = 0;
Dhlo_verifier.h214 Status HandleScatter(HloInstruction* scatter) override;
Dhlo_cost_analysis.cc877 return HandleScatter(hlo); in HandleFusion()
1077 Status HloCostAnalysis::HandleScatter(const HloInstruction* hlo) { in HandleScatter() function in xla::HloCostAnalysis
Ddynamic_dimension_inference.cc165 Status HandleScatter(HloInstruction* hlo) override;
1626 Status DynamicDimensionInferenceVisitor::HandleScatter(HloInstruction* hlo) { in HandleScatter() function in xla::DynamicDimensionInferenceVisitor
Dhlo_verifier.cc1662 Status ShapeVerifier::HandleScatter(HloInstruction* scatter) { in HandleScatter() function in xla::ShapeVerifier
Dhlo_evaluator.cc2839 Status HloEvaluator::HandleScatter(HloInstruction* hlo) { in HandleScatter() function in xla::HloEvaluator
Dhlo_instruction.cc3688 return visitor->HandleScatter(this); in Visit()
Dalgebraic_simplifier.cc5963 Status AlgebraicSimplifierVisitor::HandleScatter(HloInstruction* hlo) { in HandleScatter() function in xla::AlgebraicSimplifierVisitor
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h175 Status HandleScatter(HloInstruction* scatter) override;
Dir_emitter.cc1939 Status IrEmitter::HandleScatter(HloInstruction*) { in HandleScatter() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h496 Status HandleScatter(HloInstruction* hlo) override;
Dgather_scatter_handler.cc1040 Status SpmdPartitioningVisitor::HandleScatter(HloInstruction* hlo) { in HandleScatter() function in xla::spmd::SpmdPartitioningVisitor