Searched defs:convolution (Results 1 – 16 of 16) sorted by relevance
57 const HloInstruction& convolution, in CanFoldOperandsIntoConvolution()120 auto& convolution = *pair.first; in FoldTransposeIntoConvolution() local
53 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()102 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()130 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
201 Status ConvolutionVisitor::HandleBatchGroupCount(HloInstruction* convolution) { in HandleBatchGroupCount()319 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
3177 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in precision_config() local3187 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in mutable_precision_config() local3440 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in convolution_dimension_numbers() local3451 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_convolution_dimension_numbers() local3461 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in feature_group_count() local3473 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in batch_group_count() local
88 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution()
3394 HloInstruction* convolution) { in FoldConvInputPad()3461 HloInstruction* convolution) { in FoldConvFilterPad()3527 HloInstruction* convolution) { in SimplifyConvToDot()3642 HloInstruction* convolution) { in HandleConvolution()
434 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution()
154 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
2468 auto* convolution = local
44 const HloInstruction& convolution, in PotentiallyImplementedAsEigenConvolution()
951 const HloConvolutionInstruction* convolution, in EmitElementalConvolution()1093 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
417 void convolution(int num_iters, int kernel_x, int kernel_y) { in convolution() function
637 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
367 Status IrEmitterUnnested::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
918 def convolution(inputs, function
760 def convolution( function