Home
last modified time | relevance | path

Searched defs:HandleConvolution (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconvolution_group_converter.cc421 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::__anon30281a800111::ConvolutionVisitor
H A Ddfs_hlo_visitor_with_default.h84 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution() function
H A Dhlo_cost_analysis.cc711 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc210 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::ShapeVerifier
H A Ddynamic_dimension_inference.cc590 Status DynamicDimensionInferenceVisitor::HandleConvolution( in HandleConvolution() function in xla::DynamicDimensionInferenceVisitor
H A Dalgebraic_simplifier.cc6664 Status AlgebraicSimplifierVisitor::HandleConvolution( in HandleConvolution() function in xla::AlgebraicSimplifierVisitor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter.cc531 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::gpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc989 Status SpmdPartitioningVisitor::HandleConvolution(HloInstruction* hlo) { in HandleConvolution() function in xla::spmd::SpmdPartitioningVisitor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc905 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::cpu::IrEmitter