Home
last modified time | relevance | path

Searched defs:HandleDot (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_rewriter.cc129 Status HandleDot(HloInstruction *instr) override { in HandleDot() function in xla::gpu::__anon664e6d710111::GemmRewriterVisitor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddfs_hlo_visitor_with_default.h81 Status HandleDot(HloInstructionPtr dot) override { in HandleDot() function
H A Dhlo_cost_analysis.cc333 Status HloCostAnalysis::HandleDot(const HloInstruction* dot) { in HandleDot() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc200 Status ShapeVerifier::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::ShapeVerifier
H A Ddynamic_dimension_inference.cc491 Status DynamicDimensionInferenceVisitor::HandleDot(HloInstruction* hlo) { in HandleDot() function in xla::DynamicDimensionInferenceVisitor
H A Dalgebraic_simplifier.cc2612 Status AlgebraicSimplifierVisitor::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::AlgebraicSimplifierVisitor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc869 Status IrEmitter::HandleDot(HloInstruction* dot) { in HandleDot() function in xla::cpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc51 Status SpmdPartitioningVisitor::HandleDot(HloInstruction* hlo) { in HandleDot() function in xla::spmd::SpmdPartitioningVisitor