Home
last modified time | relevance | path

Searched refs:HandleDot (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h81 Status HandleDot(HloInstruction* dot) override;
Dir_emitter_unnested.h165 Status HandleDot(HloInstruction* dot) override;
Dir_emitter.cc467 Status IrEmitter::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc352 Status IrEmitterUnnested::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::gpu::IrEmitterUnnested
359 return IrEmitter::HandleDot(dot); in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc46 Status HandleDot(HloInstruction* hlo) override;
200 Status DynamicDimensionInferenceVisitor::HandleDot(HloInstruction* hlo) { in HandleDot() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.h71 Status HandleDot(const HloInstruction* dot) override;
Ddfs_hlo_visitor_with_default.h85 Status HandleDot(HloInstructionPtr dot) override { in HandleDot() function
Dhlo_verifier.h52 Status HandleDot(HloInstruction* dot) override;
Ddfs_hlo_visitor.h102 virtual Status HandleDot(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc239 Status HloCostAnalysis::HandleDot(const HloInstruction* dot) { in HandleDot() function in xla::HloCostAnalysis
Dhlo_evaluator_typed_visitor.h1204 Status HandleDot(HloInstruction* dot) override { in HandleDot() function
1207 return HandleDot<ReturnT>(dot); in HandleDot()
1214 Status HandleDot(HloInstruction* dot) { in HandleDot() function
1271 Status HandleDot(HloInstruction* dot) { in HandleDot() function
Dhlo_verifier.cc146 Status ShapeVerifier::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc206 Status HandleDot(HloInstruction* dot) override;
1631 Status AlgebraicSimplifierVisitor::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction.cc2536 return visitor->HandleDot(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h154 Status HandleDot(HloInstruction* dot) override;
Dir_emitter.cc915 Status IrEmitter::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::cpu::IrEmitter