| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 269 new_conv->convolution_dimension_numbers() in TEST_F()
272 new_conv->convolution_dimension_numbers() in TEST_F()
326 new_conv->convolution_dimension_numbers() in TEST_F()
329 new_conv->convolution_dimension_numbers() in TEST_F()
333 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(0)); in TEST_F()
336 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(1)); in TEST_F()
388 new_conv->convolution_dimension_numbers().input_batch_dimension()); in TEST_F()
391 new_conv->convolution_dimension_numbers().input_feature_dimension()); in TEST_F()
394 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0)); in TEST_F()
397 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1)); in TEST_F()
[all …]
|
| D | space_to_batch_converter_test.cc | 62 ->convolution_dimension_numbers() in TEST_F()
170 ->convolution_dimension_numbers() in TEST_F()
205 ->convolution_dimension_numbers() in TEST_F()
|
| D | convolution_4d_expander.cc | 40 instruction->convolution_dimension_numbers(); in InstructionMatchesPattern()
60 instruction->convolution_dimension_numbers(); in ExpandInstruction()
|
| D | result_caster.cc | 35 instruction->window(), instruction->convolution_dimension_numbers(), in MaybeInferShape()
|
| D | operand_upcaster.cc | 35 instruction->window(), instruction->convolution_dimension_numbers(), in MaybeInferShape()
|
| D | dot_as_convolution_util.cc | 81 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo()
139 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvForDotGeneralConvolution()
|
| D | space_to_batch_converter.cc | 220 return convolution->convolution_dimension_numbers() in get_chosen_spatial_dim()
337 convolution->convolution_dimension_numbers(); in IsConvSuitableForSpaceToBatch()
974 if (!are_conv_dims_compatible(consumer->convolution_dimension_numbers(), in CanPropagate()
980 if (consumer->convolution_dimension_numbers().input_batch_dimension() != in CanPropagate()
1047 kernel->shape().dimensions(consumer->convolution_dimension_numbers() in CanPropagate()
1074 if (!are_conv_dims_compatible(consumer->convolution_dimension_numbers(), in CanPropagate()
1089 consumer->convolution_dimension_numbers().input_feature_dimension()); in CanPropagate()
1108 if (!are_conv_dims_compatible(consumer->convolution_dimension_numbers(), in CanPropagate()
1154 if (!are_conv_dims_compatible(consumer->convolution_dimension_numbers(), in CanPropagate()
1159 if (!are_conv_dims_compatible(consumer->convolution_dimension_numbers(), in CanPropagate()
[all …]
|
| D | hlo_cse.cc | 144 hash, proto_hash(instruction->convolution_dimension_numbers())), in CseHash()
|
| D | dynamic_padder.cc | 240 inst->convolution_dimension_numbers().input_batch_dimension() == in ShouldSkipPadOnOperand()
910 auto dnums = custom_call_conv->convolution_dimension_numbers(); in RewriteDynamicConvolutionInputGrad()
963 custom_call_conv->convolution_dimension_numbers(), in RewriteDynamicConvolutionInputGrad()
981 auto dnums = custom_call_conv->convolution_dimension_numbers(); in RewriteDynamicConvolutionForward()
1025 custom_call_conv->convolution_dimension_numbers(), in RewriteDynamicConvolutionForward()
1043 auto dnums = custom_call_conv->convolution_dimension_numbers(); in RewriteDynamicConvolutionKernelGrad()
1110 custom_call_conv->convolution_dimension_numbers(), in RewriteDynamicConvolutionKernelGrad()
|
| D | transpose_folding.cc | 157 convolution.convolution_dimension_numbers(); in FoldTransposeIntoConvolution()
|
| D | convolution_group_converter.cc | 201 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleBatchGroupCount()
432 convolution->convolution_dimension_numbers(); in HandleConvolution()
|
| D | dynamic_dimension_inference.cc | 276 hlo->convolution_dimension_numbers(); in HandleCustomCall()
568 conv->convolution_dimension_numbers(); in HandleConvolution()
714 hlo->convolution_dimension_numbers(); in HandleDynamicConvolutionForward()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_vectorize_convolutions_test.cc | 85 conv->convolution_dimension_numbers(); in TEST_F()
138 conv->convolution_dimension_numbers(); in TEST_F()
191 conv->convolution_dimension_numbers(); in TEST_F()
244 conv->convolution_dimension_numbers(); in TEST_F()
474 conv->convolution_dimension_numbers(); in TEST_F()
542 conv->convolution_dimension_numbers(); in TEST_F()
610 conv->convolution_dimension_numbers(); in TEST_F()
|
| D | gpu_conv_padding_legalization.cc | 159 &new_conv_window, conv->convolution_dimension_numbers(), in CanonicalizeForwardConvolution()
162 MaybePaddedKernel(new_conv_window, conv->convolution_dimension_numbers(), in CanonicalizeForwardConvolution()
172 conv->convolution_dimension_numbers().kernel_spatial_dimensions(i))); in CanonicalizeForwardConvolution()
224 backward_conv->convolution_dimension_numbers(); in CanonicalizeBackwardFilterConvolution()
283 backward_conv->convolution_dimension_numbers(); in CanonicalizeBackwardInputConvolution()
|
| D | gpu_conv_rewriter.cc | 72 auto dim_numbers = conv->convolution_dimension_numbers(); in ConvertBatchGroupedToFeatureGroupedConvolution()
123 conv->convolution_dimension_numbers(); in CanImplementAsGpuForwardConv()
180 conv->convolution_dimension_numbers(); in MatchBackwardFilter()
355 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput()
531 auto conv_dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput()
711 conv->window(), conv->convolution_dimension_numbers(), in CreateCustomCallHelper()
|
| D | cudnn_vectorize_convolutions.cc | 260 const auto& dnums = conv->convolution_dimension_numbers(); in TryRevectorizeConv()
333 /*dnums=*/conv->convolution_dimension_numbers()); in TryRevectorizeConv()
368 const auto& dnums = conv->convolution_dimension_numbers(); in TryVectorizeConv()
|
| D | cudnn_fused_conv_rewriter.cc | 171 conv->convolution_dimension_numbers().output_feature_dimension()) { in FindConvWithRelu()
225 conv->convolution_dimension_numbers().output_feature_dimension()); in TryRewriteToCudnnForwardRelu()
243 conv->convolution_dimension_numbers()); in TryRewriteToCudnnForwardRelu()
|
| D | gpu_layout_assignment.cc | 71 instr->convolution_dimension_numbers(); in HeuristicLayoutAssignment()
182 instr->convolution_dimension_numbers(), input, filter, output)); in AddBackendConstraintsToDnnConvCustomCall()
|
| D | cudnn_pad_for_convolutions.cc | 174 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForTensorCore()
303 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForIntegerConvolution()
|
| D | gpu_conv_runner.cc | 488 descriptor.dnums = cudnn_call->convolution_dimension_numbers(); in GetGpuConvConfig()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 84 convolution.convolution_dimension_numbers(); in PotentiallyImplementedAsEigenConvolution()
|
| D | conv_canonicalization.cc | 40 hlo->convolution_dimension_numbers(); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| 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()
526 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
748 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionTiledOutput()
899 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvConvolution()
|
| D | dot_handler.cc | 148 original_dot->convolution_dimension_numbers(); in GenNewWindow()
235 original_dot->convolution_dimension_numbers(); in GenNewConvDNums()
2438 const auto& dnums = original_hlo->convolution_dimension_numbers(); in ConvertDimsMappingWithFeatureGroupCount()
2461 const auto& dnums = original_hlo->convolution_dimension_numbers(); in ConvertDimsMappingWithBatchGroupCount()
2905 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionDot()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 535 const auto& dim_nums = conv->convolution_dimension_numbers(); in EmitConvolutionForwardAsMlir()
|