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.h92 Status HandleScatter(HloInstruction* scatter) override;
Dir_emitter_unnested.h201 Status HandleScatter(HloInstruction* scatter) override;
Dir_emitter.cc211 Status IrEmitter::HandleScatter(HloInstruction*) { in HandleScatter() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc2491 Status IrEmitterUnnested::HandleScatter(HloInstruction* scatter) { in HandleScatter() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h123 Status HandleScatter(const HloInstruction* scatter) override;
Ddfs_hlo_visitor_with_default.h237 Status HandleScatter(HloInstructionPtr scatter) override { in HandleScatter() function
Dhlo_verifier.h109 Status HandleScatter(HloInstruction* scatter) override;
Ddfs_hlo_visitor.h264 virtual Status HandleScatter(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc822 return HandleScatter(hlo); in HandleFusion()
1019 Status HloCostAnalysis::HandleScatter(const HloInstruction* scatter) { in HandleScatter() function in xla::HloCostAnalysis
Ddynamic_dimension_inference.cc149 Status HandleScatter(HloInstruction* hlo) override;
1404 Status DynamicDimensionInferenceVisitor::HandleScatter(HloInstruction* hlo) { in HandleScatter() function in xla::DynamicDimensionInferenceVisitor
Dhlo_verifier.cc1139 Status ShapeVerifier::HandleScatter(HloInstruction* scatter) { in HandleScatter() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc359 Status HandleScatter(HloInstruction* scatter) override;
5011 Status AlgebraicSimplifierVisitor::HandleScatter(HloInstruction* scatter) { in HandleScatter() function in xla::AlgebraicSimplifierVisitor
Dhlo_evaluator_typed_visitor.h2330 Status HandleScatter(HloInstruction* scatter) override {
Dhlo_instruction.cc3255 return visitor->HandleScatter(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h167 Status HandleScatter(HloInstruction* scatter) override;
Dir_emitter.cc1828 Status IrEmitter::HandleScatter(HloInstruction*) { in HandleScatter() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h433 Status HandleScatter(HloInstruction* hlo) override;
Dgather_scatter_handler.cc507 Status SpmdPartitioningVisitor::HandleScatter(HloInstruction* hlo) { in HandleScatter() function in xla::spmd::SpmdPartitioningVisitor