/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | transpose_folding_test.cc | 276 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 D | space_to_batch_converter_test.cc | 64 ->convolution_dimension_numbers() in TEST_F() 207 ->convolution_dimension_numbers() in TEST_F() 244 ->convolution_dimension_numbers() in TEST_F()
|
H A D | space_to_batch_converter.cc | 245 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 D | convolution_4d_expander.cc | 40 instruction->convolution_dimension_numbers(); in InstructionMatchesPattern() 60 instruction->convolution_dimension_numbers(); in ExpandInstruction()
|
H A D | dynamic_padder.cc | 239 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 D | dot_as_convolution_util.cc | 82 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo() 140 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvForDotGeneralConvolution()
|
H A D | result_caster.cc | 35 instruction->window(), instruction->convolution_dimension_numbers(), in MaybeInferShape()
|
H A D | operand_upcaster.cc | 35 instruction->window(), instruction->convolution_dimension_numbers(), in MaybeInferShape()
|
H A D | convolution_group_converter.cc | 205 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleBatchGroupCount() 437 convolution->convolution_dimension_numbers(); in HandleConvolution()
|
H A D | transpose_folding.cc | 124 convolution.convolution_dimension_numbers(); in FoldTransposeIntoConvolution()
|
H A D | hlo_cse.cc | 179 instruction->convolution_dimension_numbers(); in AbslHashValue()
|
H A D | dynamic_dimension_inference.cc | 305 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 D | cudnn_vectorize_convolutions_test.cc | 85 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 D | gpu_conv_padding_legalization.cc | 162 &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 D | gpu_conv_rewriter.cc | 44 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 D | cudnn_simplify_padding.cc | 59 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 D | cudnn_vectorize_convolutions.cc | 266 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 D | cudnn_fused_conv_rewriter.cc | 123 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 D | gpu_layout_assignment.cc | 76 instr->convolution_dimension_numbers(); in HeuristicLayoutAssignment() 187 instr->convolution_dimension_numbers(), input, filter, output)); in AddBackendConstraintsToDnnConvCustomCall()
|
H A D | cudnn_pad_for_convolutions.cc | 184 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForTensorCore() 314 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForIntegerConvolution()
|
H A D | cudnn_support_utils.cc | 34 const auto& dnums = conv.convolution_dimension_numbers(); in CudnnSupportsOptimizedIntegerConvolution()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
H A D | convolution_handler.cc | 55 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 D | dot_handler.cc | 157 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 D | ir_emission_utils.cc | 83 convolution.convolution_dimension_numbers(); in PotentiallyImplementedAsEigenConvolution()
|
H A D | conv_canonicalization.cc | 42 hlo->convolution_dimension_numbers(); in Run()
|