Home
last modified time | relevance | path

Searched refs:convolution_dimension_numbers (Results 1 – 25 of 42) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtranspose_folding_test.cc276 new_conv->convolution_dimension_numbers() in TEST_F()
279 new_conv->convolution_dimension_numbers() in TEST_F()
334 new_conv->convolution_dimension_numbers() in TEST_F()
337 new_conv->convolution_dimension_numbers() in TEST_F()
341 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(0)); in TEST_F()
344 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(1)); in TEST_F()
397 new_conv->convolution_dimension_numbers().input_batch_dimension()); in TEST_F()
400 new_conv->convolution_dimension_numbers().input_feature_dimension()); in TEST_F()
403 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0)); in TEST_F()
406 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1)); in TEST_F()
[all …]
H A Dspace_to_batch_converter_test.cc64 ->convolution_dimension_numbers() in TEST_F()
207 ->convolution_dimension_numbers() in TEST_F()
244 ->convolution_dimension_numbers() in TEST_F()
H A Dspace_to_batch_converter.cc245 const int64_t end_point = convolution->convolution_dimension_numbers() in GetFirstChosenSpatialDim()
257 convolution->convolution_dimension_numbers().input_spatial_dimensions( in GetChosenSpatialDims()
397 convolution->convolution_dimension_numbers(); in IsConvSuitableForSpaceToBatch()
480 auto dim_numbers = convolution->convolution_dimension_numbers(); in IsThisBackPropFilterConv()
491 if (convolution->convolution_dimension_numbers() in IsThisBackPropFilterConv()
1125 if (!are_conv_dims_compatible(consumer->convolution_dimension_numbers(), in CanPropagate()
1131 if (consumer->convolution_dimension_numbers().input_batch_dimension() != in CanPropagate()
1199 kernel->shape().dimensions(consumer->convolution_dimension_numbers() in CanPropagate()
1228 if (!are_conv_dims_compatible(consumer->convolution_dimension_numbers(), in CanPropagate()
1243 consumer->convolution_dimension_numbers().input_feature_dimension()); in CanPropagate()
[all …]
H A Dconvolution_4d_expander.cc40 instruction->convolution_dimension_numbers(); in InstructionMatchesPattern()
60 instruction->convolution_dimension_numbers(); in ExpandInstruction()
H A Ddynamic_padder.cc239 inst->convolution_dimension_numbers().input_batch_dimension()) { in ShouldSkipPadOnOperand()
243 inst->convolution_dimension_numbers().input_spatial_dimensions(); in ShouldSkipPadOnOperand()
255 (dimension == inst->convolution_dimension_numbers() in ShouldSkipPadOnOperand()
966 auto dnums = custom_call_conv->convolution_dimension_numbers(); in RewriteDynamicConvolutionInputGrad()
1021 custom_call_conv->convolution_dimension_numbers(), in RewriteDynamicConvolutionInputGrad()
1037 auto dnums = custom_call_conv->convolution_dimension_numbers(); in RewriteDynamicConvolutionForward()
1082 custom_call_conv->convolution_dimension_numbers(), in RewriteDynamicConvolutionForward()
1098 auto dnums = custom_call_conv->convolution_dimension_numbers(); in RewriteDynamicConvolutionKernelGrad()
1166 custom_call_conv->convolution_dimension_numbers(), in RewriteDynamicConvolutionKernelGrad()
H A Ddot_as_convolution_util.cc82 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo()
140 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvForDotGeneralConvolution()
H A Dresult_caster.cc35 instruction->window(), instruction->convolution_dimension_numbers(), in MaybeInferShape()
H A Doperand_upcaster.cc35 instruction->window(), instruction->convolution_dimension_numbers(), in MaybeInferShape()
H A Dconvolution_group_converter.cc205 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleBatchGroupCount()
437 convolution->convolution_dimension_numbers(); in HandleConvolution()
H A Dtranspose_folding.cc124 convolution.convolution_dimension_numbers(); in FoldTransposeIntoConvolution()
H A Dhlo_cse.cc179 instruction->convolution_dimension_numbers(); in AbslHashValue()
H A Ddynamic_dimension_inference.cc305 hlo->convolution_dimension_numbers(); in HandleCustomCall()
597 conv->convolution_dimension_numbers(); in HandleConvolution()
743 hlo->convolution_dimension_numbers(); in HandleDynamicConvolutionForward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_vectorize_convolutions_test.cc85 conv->convolution_dimension_numbers(); in TEST_F()
158 conv->convolution_dimension_numbers(); in TEST_F()
211 conv->convolution_dimension_numbers(); in TEST_F()
264 conv->convolution_dimension_numbers(); in TEST_F()
494 conv->convolution_dimension_numbers(); in TEST_F()
562 conv->convolution_dimension_numbers(); in TEST_F()
630 conv->convolution_dimension_numbers(); in TEST_F()
H A Dgpu_conv_padding_legalization.cc162 &new_conv_window, conv->convolution_dimension_numbers(), in CanonicalizeForwardConvolution()
165 MaybePaddedKernel(new_conv_window, conv->convolution_dimension_numbers(), in CanonicalizeForwardConvolution()
175 conv->convolution_dimension_numbers().kernel_spatial_dimensions(i))); in CanonicalizeForwardConvolution()
227 backward_conv->convolution_dimension_numbers(); in CanonicalizeBackwardFilterConvolution()
286 backward_conv->convolution_dimension_numbers(); in CanonicalizeBackwardInputConvolution()
H A Dgpu_conv_rewriter.cc44 conv->convolution_dimension_numbers(); in CanImplementAsGpuForwardConv()
101 conv->convolution_dimension_numbers(); in MatchBackwardFilter()
276 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput()
452 auto conv_dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput()
608 auto dim_numbers = conv->convolution_dimension_numbers(); in ConvertBatchGroupedToFeatureGroupedConvolution()
695 conv->window(), conv->convolution_dimension_numbers(), in CreateCustomCallHelper()
H A Dcudnn_simplify_padding.cc59 conv->convolution_dimension_numbers(); in FindOutputVectCDim()
73 conv->convolution_dimension_numbers(); in FindKernelVectCDim()
91 conv->convolution_dimension_numbers(); in NumTrailingZeroOutputFeatures()
272 const auto& dnums = conv->convolution_dimension_numbers(); in TrySimplifyPadding()
H A Dcudnn_vectorize_convolutions.cc266 const auto& dnums = conv->convolution_dimension_numbers(); in TryRevectorizeConv()
354 /*dnums=*/conv->convolution_dimension_numbers()); in TryRevectorizeConv()
403 const auto& dnums = conv->convolution_dimension_numbers(); in TryVectorizeConv()
H A Dcudnn_fused_conv_rewriter.cc123 conv->convolution_dimension_numbers().output_feature_dimension()); in EnsureIsConvBiasActivation()
286 conv->convolution_dimension_numbers().output_feature_dimension() && in FuseBiasOrSideInput()
304 {gte->shape().dimensions(conv->convolution_dimension_numbers() in FuseBiasOrSideInput()
H A Dgpu_layout_assignment.cc76 instr->convolution_dimension_numbers(); in HeuristicLayoutAssignment()
187 instr->convolution_dimension_numbers(), input, filter, output)); in AddBackendConstraintsToDnnConvCustomCall()
H A Dcudnn_pad_for_convolutions.cc184 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForTensorCore()
314 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForIntegerConvolution()
H A Dcudnn_support_utils.cc34 const auto& dnums = conv.convolution_dimension_numbers(); in CudnnSupportsOptimizedIntegerConvolution()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc55 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithBatchGroupCount()
147 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithFeatureGroupCount()
239 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
528 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
750 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionTiledOutput()
901 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvConvolution()
H A Ddot_handler.cc157 original_dot->convolution_dimension_numbers(); in GenNewWindow()
244 original_dot->convolution_dimension_numbers(); in GenNewConvDNums()
2798 const auto& dnums = original_hlo->convolution_dimension_numbers(); in ConvertDimsMappingWithFeatureGroupCount()
2821 const auto& dnums = original_hlo->convolution_dimension_numbers(); in ConvertDimsMappingWithBatchGroupCount()
3430 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionDot()
3455 original_hlo->convolution_dimension_numbers() in PartitionDot()
3458 original_hlo->convolution_dimension_numbers() in PartitionDot()
3479 original_hlo->convolution_dimension_numbers() in PartitionDot()
3482 original_hlo->convolution_dimension_numbers() in PartitionDot()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emission_utils.cc83 convolution.convolution_dimension_numbers(); in PotentiallyImplementedAsEigenConvolution()
H A Dconv_canonicalization.cc42 hlo->convolution_dimension_numbers(); in Run()

12