Searched defs:convolution (Results 1 – 17 of 17) sorted by relevance
28 ### Change FlatBuffer schema70 ### Change C structures and kernel implementation108 ### Change the FlatBuffer reading code153 ### Change kernel registration182 ### Change TFLite op version206 ### Update the operator version map224 ### Delegation implementation
85 const HloInstruction* convolution) { in GetConvolutionFlops()
531 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
42 const HloInstruction& convolution, in CanFoldOperandsIntoConvolution()116 auto& convolution = *pair.first; in FoldTransposeIntoConvolution() local
243 int64_t GetFirstChosenSpatialDim(HloInstruction* convolution) { in GetFirstChosenSpatialDim()251 std::vector<int64_t> GetChosenSpatialDims(HloInstruction* convolution) { in GetChosenSpatialDims()341 auto convolution = inst; in ConvolutionVisitor() local373 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in IsForwardWindowDilatedConv()395 HloInstruction* convolution) { in IsConvSuitableForSpaceToBatch()477 bool ConvolutionVisitor::IsThisBackPropFilterConv(HloInstruction* convolution) { in IsThisBackPropFilterConv()2536 Status ConvolutionVisitor::PropagateOnConv(HloInstruction* convolution) { in PropagateOnConv()2995 HloInstruction* convolution) { in PropagateOnBackpropFilterConv()3543 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in GetConvolutionDetails()3626 HloInstruction* convolution) { in PerformSpaceToBatchOnConvolution()
204 Status ConvolutionVisitor::HandleBatchGroupCount(HloInstruction* convolution) { in HandleBatchGroupCount()421 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
580 const HloInstruction* convolution) { in GetConvolutionFlops()591 int64_t HloCostAnalysis::GetConvolutionFlops(const HloInstruction* convolution, in GetConvolutionFlops()711 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution()
4391 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in precision_config() local4405 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in mutable_precision_config() local4719 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in convolution_dimension_numbers() local4730 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_convolution_dimension_numbers() local4740 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in feature_group_count() local4747 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_feature_group_count() local4755 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in batch_group_count() local4762 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_batch_group_count() local
84 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution()
6279 HloInstruction* convolution) { in FoldConvInputPad()6340 HloInstruction* convolution) { in FoldConvFilterPad()6406 HloInstruction* convolution) { in SwapConvOperands()6543 HloInstruction* convolution) { in SimplifyConvToDot()6665 HloInstruction* convolution) { in HandleConvolution()
210 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
3486 auto* convolution = local
3022 const HloInstruction* convolution, in EmitConvolution()
43 const HloInstruction& convolution, in PotentiallyImplementedAsEigenConvolution()
905 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
492 void convolution(int num_iters, int kernel_x, int kernel_y) { in convolution() function
1030 def convolution( function