Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_rewriter.cc49 Status HandleDot(HloInstruction *instr) override { in HandleDot() function in xla::gpu::GemmRewriterVisitor
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h74 Status HandleDot(const HloInstruction* dot) override;
Ddfs_hlo_visitor_with_default.h86 Status HandleDot(HloInstructionPtr dot) override { in HandleDot() function
Dhlo_verifier.h54 Status HandleDot(HloInstruction* dot) override;
Ddfs_hlo_visitor.h102 virtual Status HandleDot(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc95 Status HandleDot(HloInstruction* hlo) override;
459 Status DynamicDimensionInferenceVisitor::HandleDot(HloInstruction* hlo) { in HandleDot() function in xla::DynamicDimensionInferenceVisitor
Dhlo_evaluator_typed_visitor.h1322 Status HandleDot(HloInstruction* dot) override {
1327 return HandleDot<ReturnT>(dot);
1334 Status HandleDot(HloInstruction* dot) {
1391 Status HandleDot(HloInstruction* dot) {
Dhlo_cost_analysis.cc323 Status HloCostAnalysis::HandleDot(const HloInstruction* dot) { in HandleDot() function in xla::HloCostAnalysis
Dhlo_verifier.cc140 Status ShapeVerifier::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc313 Status HandleDot(HloInstruction* dot) override;
2305 Status AlgebraicSimplifierVisitor::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction.cc3117 return visitor->HandleDot(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h139 Status HandleDot(HloInstruction* dot) override;
Dir_emitter.cc834 Status IrEmitter::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h413 Status HandleDot(HloInstruction* hlo) override;
Ddot_handler.cc42 Status SpmdPartitioningVisitor::HandleDot(HloInstruction* hlo) { in HandleDot() function in xla::spmd::SpmdPartitioningVisitor