Home
last modified time | relevance | path

Searched defs:HandleDot (Results 1 – 8 of 8) 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/
Ddfs_hlo_visitor_with_default.h86 Status HandleDot(HloInstructionPtr dot) override { in HandleDot() function
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
Ddynamic_dimension_inference.cc459 Status DynamicDimensionInferenceVisitor::HandleDot(HloInstruction* hlo) { in HandleDot() function in xla::DynamicDimensionInferenceVisitor
Dalgebraic_simplifier.cc2305 Status AlgebraicSimplifierVisitor::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::AlgebraicSimplifierVisitor
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc42 Status SpmdPartitioningVisitor::HandleDot(HloInstruction* hlo) { in HandleDot() function in xla::spmd::SpmdPartitioningVisitor
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc834 Status IrEmitter::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::cpu::IrEmitter