Searched defs:convolution (Results 1 – 18 of 18) sorted by relevance
28 ### Change FlatBuffer schema70 ### Change C structures and kernel implementation108 ### Change the FlatBuffer reading code153 ### Change kernel registration182 ### Change TFLite op version231 ### Update the operator version map249 ### Delegation implementation
86 const HloInstruction& convolution, in CanFoldOperandsIntoConvolution()149 auto& convolution = *pair.first; in FoldTransposeIntoConvolution() local
52 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()103 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()131 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
193 int64 get_chosen_spatial_dim(HloInstruction* convolution) { in get_chosen_spatial_dim()262 auto convolution = inst; in ConvolutionVisitor() local277 HloInstruction* convolution) { in IsConvSuitableForSpaceToBatch()1845 Status ConvolutionVisitor::PropagateOnConv(HloInstruction* convolution) { in PropagateOnConv()2147 HloInstruction* convolution) { in PropagateOnBackpropFilterConv()2592 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in GetConvolutionDetails()2653 HloInstruction* convolution) { in PerformSpaceToBatchOnConvolution()
200 Status ConvolutionVisitor::HandleBatchGroupCount(HloInstruction* convolution) { in HandleBatchGroupCount()416 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
3853 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in precision_config() local3867 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in mutable_precision_config() local4146 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in convolution_dimension_numbers() local4157 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_convolution_dimension_numbers() local4167 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in feature_group_count() local4174 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_feature_group_count() local4182 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in batch_group_count() local4189 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_batch_group_count() local
89 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution()
5143 HloInstruction* convolution) { in FoldConvInputPad()5210 HloInstruction* convolution) { in FoldConvFilterPad()5276 HloInstruction* convolution) { in SwapConvOperands()5400 HloInstruction* convolution) { in SimplifyConvToDot()5518 HloInstruction* convolution) { in HandleConvolution()
569 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution()
150 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
2980 auto* convolution = local
2819 const HloInstruction* convolution, in EmitConvolution()
44 const HloInstruction& convolution, in PotentiallyImplementedAsEigenConvolution()
870 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
417 void convolution(int num_iters, int kernel_x, int kernel_y) { in convolution() function
549 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
806 Status IrEmitterUnnested::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
866 def convolution( function