Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h94 Status HandleScatter(HloInstruction* scatter) override;
Dir_emitter_unnested.h176 Status HandleScatter(HloInstruction* scatter) override;
Dir_emitter.cc138 Status IrEmitter::HandleScatter(HloInstruction*) { in HandleScatter() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc1029 Status IrEmitterUnnested::HandleScatter(HloInstruction* scatter) { in HandleScatter() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h113 Status HandleScatter(const HloInstruction* scatter) override;
Ddfs_hlo_visitor_with_default.h209 Status HandleScatter(HloInstructionPtr scatter) override { in HandleScatter() function
Dhlo_verifier.h98 Status HandleScatter(HloInstruction* scatter) override;
Ddfs_hlo_visitor.h247 virtual Status HandleScatter(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc720 Status HloCostAnalysis::HandleScatter(const HloInstruction* scatter) { in HandleScatter() function in xla::HloCostAnalysis
Dhlo_verifier.cc818 Status ShapeVerifier::HandleScatter(HloInstruction* scatter) { in HandleScatter() function in xla::ShapeVerifier
Dhlo_evaluator_typed_visitor.h2307 Status HandleScatter(HloInstruction* scatter) override { in HandleScatter() function
Dhlo_instruction.cc2652 return visitor->HandleScatter(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h181 Status HandleScatter(HloInstruction* scatter) override;
Dir_emitter.cc1914 Status IrEmitter::HandleScatter(HloInstruction*) { in HandleScatter() function in xla::cpu::IrEmitter