Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h67 Status HandleDot(HloInstruction* dot) override;
Dir_emitter.h76 Status HandleDot(HloInstruction* dot) override;
Dir_emitter.cc441 Status IrEmitter::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc293 Status IrEmitterUnnested::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::gpu::IrEmitterUnnested
304 return IrEmitter::HandleDot(dot); in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.h42 Status HandleDot(HloInstruction* dot) override;
Dhlo_cost_analysis.h68 Status HandleDot(const HloInstruction* dot) override;
Ddfs_hlo_visitor_with_default.h82 Status HandleDot(HloInstructionPtr dot) override { in HandleDot() function
Ddfs_hlo_visitor.h101 virtual Status HandleDot(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc201 Status HloCostAnalysis::HandleDot(const HloInstruction* dot) { in HandleDot() function in xla::HloCostAnalysis
Dhlo_verifier.cc67 Status ShapeVerifier::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc145 Status HandleDot(HloInstruction* dot) override;
920 Status AlgebraicSimplifierVisitor::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::AlgebraicSimplifierVisitor
Dhlo_evaluator.cc1026 Status HandleDot(HloInstruction* dot) override { in HandleDot() function in xla::HloEvaluator::TypedVisitor
Dhlo_instruction.cc2579 return visitor->HandleDot(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h125 Status HandleDot(HloInstruction* dot) override;
Dir_emitter.cc799 Status IrEmitter::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::cpu::IrEmitter