/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 | 61 ->convolution_dimension_numbers() in TEST_F() 110 ->convolution_dimension_numbers() in TEST_F() 144 ->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 | operand_upcaster.cc | 35 instruction->window(), instruction->convolution_dimension_numbers(), in MaybeInferShape()
|
D | dot_as_convolution_util.cc | 55 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo() 113 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvForDotGeneralConvolution()
|
D | dynamic_padder.cc | 230 inst->convolution_dimension_numbers().input_batch_dimension() == in ShouldSkipPadOnOperand() 812 auto dnums = custom_call_conv->convolution_dimension_numbers(); in RewriteDynamicConvolutionInputGrad() 864 custom_call_conv->convolution_dimension_numbers(), in RewriteDynamicConvolutionInputGrad() 882 auto dnums = custom_call_conv->convolution_dimension_numbers(); in RewriteDynamicConvolutionForward() 925 custom_call_conv->convolution_dimension_numbers(), in RewriteDynamicConvolutionForward() 943 auto dnums = custom_call_conv->convolution_dimension_numbers(); in RewriteDynamicConvolutionKernelGrad() 1009 custom_call_conv->convolution_dimension_numbers(), in RewriteDynamicConvolutionKernelGrad()
|
D | hlo_cse.cc | 135 hash, proto_hash(instruction->convolution_dimension_numbers())), in CseHash()
|
D | space_to_batch_converter.cc | 194 return convolution->convolution_dimension_numbers() in get_chosen_spatial_dim() 279 convolution->convolution_dimension_numbers(); in IsConvSuitableForSpaceToBatch() 766 if (consumer->convolution_dimension_numbers().input_spatial_dimensions( in CanPropagate() 772 if (consumer->convolution_dimension_numbers().input_batch_dimension() != in CanPropagate() 823 kernel->shape().dimensions(consumer->convolution_dimension_numbers() in CanPropagate() 859 consumer->convolution_dimension_numbers().input_feature_dimension()); in CanPropagate() 1852 auto original_conv_dims = convolution->convolution_dimension_numbers(); in PropagateOnConv() 2155 auto original_conv_dims = convolution->convolution_dimension_numbers(); in PropagateOnBackpropFilterConv() 2659 convolution->convolution_dimension_numbers(); in PerformSpaceToBatchOnConvolution()
|
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 | 274 hlo->convolution_dimension_numbers(); in HandleCustomCall() 567 conv->convolution_dimension_numbers(); in HandleConvolution() 695 hlo->convolution_dimension_numbers(); in HandleDynamicConvolutionForward()
|
D | hlo_instructions.cc | 2213 convolution_dimension_numbers(), in IdenticalSlowPath() 2214 casted_other.convolution_dimension_numbers()) && in IdenticalSlowPath() 2520 convolution_dimension_numbers(), in IdenticalSlowPath() 2521 casted_other.convolution_dimension_numbers()))) { in IdenticalSlowPath()
|
D | hlo_instruction.cc | 519 proto.convolution_dimension_numbers(), precision_config); in CreateFromProto() 591 proto.convolution_dimension_numbers()); in CreateFromProto() 4145 HloInstruction::convolution_dimension_numbers() const { in convolution_dimension_numbers() function in xla::HloInstruction 4147 return convolution->convolution_dimension_numbers(); in convolution_dimension_numbers() 4150 return custom_call->convolution_dimension_numbers(); in convolution_dimension_numbers()
|
D | sharding_propagation.cc | 132 const auto& dnums = instruction->convolution_dimension_numbers(); in IsConvolutionKernelSmall() 517 const auto& dnums = instruction->convolution_dimension_numbers(); in InferConvolutionShardingFromOperands()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
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() 344 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput() 504 auto conv_dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput() 684 conv->window(), conv->convolution_dimension_numbers(), in CreateCustomCallHelper()
|
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 | cudnn_pad_for_convolutions.cc | 182 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForTensorCore() 299 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForIntegerConvolution()
|
D | gpu_layout_assignment.cc | 163 instr->convolution_dimension_numbers(), input, filter, output)); in AddBackendConstraintsToDnnConvCustomCall()
|
D | gpu_conv_runner.cc | 452 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 | 144 original_dot->convolution_dimension_numbers(); in GenNewWindow() 230 original_dot->convolution_dimension_numbers(); in GenNewConvDNums() 2249 const auto& dnums = original_hlo->convolution_dimension_numbers(); in ConvertDimsMappingWithFeatureGroupCount() 2272 const auto& dnums = original_hlo->convolution_dimension_numbers(); in ConvertDimsMappingWithBatchGroupCount() 2516 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionDot()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
D | conv_emitter.cc | 534 const auto& dim_nums = conv->convolution_dimension_numbers(); in EmitConvolutionForwardAsMlir()
|