Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding_test.cc269 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 …]
Dspace_to_batch_converter_test.cc61 ->convolution_dimension_numbers() in TEST_F()
110 ->convolution_dimension_numbers() in TEST_F()
144 ->convolution_dimension_numbers() in TEST_F()
Dconvolution_4d_expander.cc40 instruction->convolution_dimension_numbers(); in InstructionMatchesPattern()
60 instruction->convolution_dimension_numbers(); in ExpandInstruction()
Doperand_upcaster.cc35 instruction->window(), instruction->convolution_dimension_numbers(), in MaybeInferShape()
Ddot_as_convolution_util.cc55 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo()
113 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvForDotGeneralConvolution()
Ddynamic_padder.cc230 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()
Dhlo_cse.cc135 hash, proto_hash(instruction->convolution_dimension_numbers())), in CseHash()
Dspace_to_batch_converter.cc194 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()
Dtranspose_folding.cc157 convolution.convolution_dimension_numbers(); in FoldTransposeIntoConvolution()
Dconvolution_group_converter.cc201 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleBatchGroupCount()
432 convolution->convolution_dimension_numbers(); in HandleConvolution()
Ddynamic_dimension_inference.cc274 hlo->convolution_dimension_numbers(); in HandleCustomCall()
567 conv->convolution_dimension_numbers(); in HandleConvolution()
695 hlo->convolution_dimension_numbers(); in HandleDynamicConvolutionForward()
Dhlo_instructions.cc2213 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()
Dhlo_instruction.cc519 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()
Dsharding_propagation.cc132 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/
Dgpu_conv_padding_legalization.cc159 &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()
Dgpu_conv_rewriter.cc72 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()
Dcudnn_fused_conv_rewriter.cc171 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()
Dcudnn_pad_for_convolutions.cc182 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForTensorCore()
299 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForIntegerConvolution()
Dgpu_layout_assignment.cc163 instr->convolution_dimension_numbers(), input, filter, output)); in AddBackendConstraintsToDnnConvCustomCall()
Dgpu_conv_runner.cc452 descriptor.dnums = cudnn_call->convolution_dimension_numbers(); in GetGpuConvConfig()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc84 convolution.convolution_dimension_numbers(); in PotentiallyImplementedAsEigenConvolution()
Dconv_canonicalization.cc40 hlo->convolution_dimension_numbers(); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
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()
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()
Ddot_handler.cc144 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/
Dconv_emitter.cc534 const auto& dim_nums = conv->convolution_dimension_numbers(); in EmitConvolutionForwardAsMlir()

12