Home
last modified time | relevance | path

Searched refs:input_spatial_dimensions (Results 1 – 25 of 48) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_vectorize_convolutions_test.cc86 ASSERT_EQ(dnums.input_spatial_dimensions().size(), 2); in TEST_F()
91 EXPECT_EQ(dnums.input_spatial_dimensions()[0], 1); in TEST_F()
92 EXPECT_EQ(dnums.input_spatial_dimensions()[1], 2); in TEST_F()
159 ASSERT_EQ(dnums.input_spatial_dimensions().size(), 2); in TEST_F()
165 EXPECT_EQ(dnums.input_spatial_dimensions()[0], 3); in TEST_F()
166 EXPECT_EQ(dnums.input_spatial_dimensions()[1], 4); in TEST_F()
212 ASSERT_EQ(dnums.input_spatial_dimensions().size(), 2); in TEST_F()
218 EXPECT_EQ(dnums.input_spatial_dimensions()[0], 3); in TEST_F()
219 EXPECT_EQ(dnums.input_spatial_dimensions()[1], 4); in TEST_F()
265 ASSERT_EQ(dnums.input_spatial_dimensions().size(), 2); in TEST_F()
[all …]
H A Dgpu_conv_padding_legalization.cc65 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) { in MaybePaddedAndSlicedInput()
66 int64_t dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput()
100 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) { in MaybePaddedAndSlicedInput()
101 int64_t dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput()
239 int64_t dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardFilterConvolution()
323 int64_t dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()
368 int64_t dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()
H A Dstream_executor_util.cc76 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
77 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
84 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
85 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
91 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
92 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
301 dnums.input_spatial_dimensions()), in FindVectorizedFeatureDims()
H A Dcudnn_support_utils.cc70 if (dnums.input_spatial_dimensions().size() != 2 || in CudnnSupportsOptimizedIntegerConvolution()
95 int64_t W = input_shape.dimensions(dnums.input_spatial_dimensions()[0]); in CudnnSupportsOptimizedIntegerConvolution()
96 int64_t H = input_shape.dimensions(dnums.input_spatial_dimensions()[1]); in CudnnSupportsOptimizedIntegerConvolution()
H A Dgpu_conv_rewriter.cc104 auto input_spatial_dims = conv_dnums.input_spatial_dimensions(); in MatchBackwardFilter()
346 const auto& input_spatial_dims = dnums.input_spatial_dimensions(); in MatchBackwardInput()
461 conv_dnums.input_spatial_dimensions(i)); in MatchBackwardInput()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconvolution_4d_expander.cc41 if (dim_nums.input_spatial_dimensions().size() != 4) { in InstructionMatchesPattern()
45 for (int64_t i = 0; i < dim_nums.input_spatial_dimensions().size(); ++i) { in InstructionMatchesPattern()
46 int64_t spatial_dim = dim_nums.input_spatial_dimensions(i); in InstructionMatchesPattern()
74 for (int64_t i = 0; i < dim_nums.input_spatial_dimensions().size(); ++i) { in ExpandInstruction()
75 int64_t input_spatial_dim = dim_nums.input_spatial_dimensions(i); in ExpandInstruction()
132 for (int64_t i = 0; i < new_dim_nums.input_spatial_dimensions().size(); ++i) { in ExpandInstruction()
135 new_dim_nums.input_spatial_dimensions(i))); in ExpandInstruction()
H A Dtranspose_folding_test.cc402 dnums.input_spatial_dimensions(0), in TEST_F()
403 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0)); in TEST_F()
405 dnums.input_spatial_dimensions(1), in TEST_F()
406 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1)); in TEST_F()
471 dnums.input_spatial_dimensions(0), in TEST_F()
472 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1)); in TEST_F()
474 dnums.input_spatial_dimensions(1), in TEST_F()
475 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0)); in TEST_F()
H A Ddot_as_convolution_util.cc95 int64_t lhs = conv_dims.input_spatial_dimensions(i); in ParseConvolutionDimsInfo()
145 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
155 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
H A Dspace_to_batch_converter.cc257 convolution->convolution_dimension_numbers().input_spatial_dimensions( in GetChosenSpatialDims()
710 auto it = absl::c_find(dim_numbers.input_spatial_dimensions(), i); in BringSpaceNextToBatch()
711 if (it != dim_numbers.input_spatial_dimensions().end()) { in BringSpaceNextToBatch()
712 int64_t j = it - dim_numbers.input_spatial_dimensions().begin(); in BringSpaceNextToBatch()
1082 if (dim_numbers.input_spatial_dimensions( in CanPropagate()
1087 for (int i = 0; i < dim_numbers.input_spatial_dimensions().size(); in CanPropagate()
1089 if (dim_numbers.input_spatial_dimensions(i) == in CanPropagate()
1091 dim_numbers.input_spatial_dimensions(i) == in CanPropagate()
2563 original_conv_dims.input_spatial_dimensions(i))); in PropagateOnConv()
3009 old_split_spatial_dims[i] = original_conv_dims.input_spatial_dimensions( in PropagateOnBackpropFilterConv()
[all …]
H A Ddynamic_padder.cc243 inst->convolution_dimension_numbers().input_spatial_dimensions(); in ShouldSkipPadOnOperand()
977 dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad()
1005 [&](int64_t dim) { return dnums.input_spatial_dimensions(dim); }); in RewriteDynamicConvolutionInputGrad()
1047 dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionForward()
1074 [&](int64_t dim) { return dnums.input_spatial_dimensions(dim); }); in RewriteDynamicConvolutionForward()
1108 dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionKernelGrad()
1158 [&](int64_t dim) { return dnums.input_spatial_dimensions(dim); }); in RewriteDynamicConvolutionKernelGrad()
H A Dhlo_cse.cc183 absl::MakeSpan(conv_dimension_numbers.input_spatial_dimensions()), in AbslHashValue()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc74 dnums.input_spatial_dimensions(i); in PartitionConvolutionWithBatchGroupCount()
90 lhs_to_output_indices[dnums.input_spatial_dimensions(i)] = in PartitionConvolutionWithBatchGroupCount()
165 dnums.input_spatial_dimensions(i); in PartitionConvolutionWithFeatureGroupCount()
181 lhs_to_output_indices[dnums.input_spatial_dimensions(i)] = in PartitionConvolutionWithFeatureGroupCount()
247 dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
321 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
358 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
539 dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
630 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
658 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
[all …]
H A Ddot_handler.cc160 if (conv_dnums.input_spatial_dimensions(i) == lhs_concat_dim) { in GenNewWindow()
248 std::vector<int64_t> input_spatial_dimensions( in GenNewConvDNums() local
249 dnums.input_spatial_dimensions().begin(), in GenNewConvDNums()
250 dnums.input_spatial_dimensions().end()); in GenNewConvDNums()
258 for (int64_t i = 0; i < input_spatial_dimensions.size(); ++i) { in GenNewConvDNums()
259 if (lhs_concat_dim <= input_spatial_dimensions[i]) { in GenNewConvDNums()
260 input_spatial_dimensions[i]++; in GenNewConvDNums()
263 input_spatial_dimensions.push_back(lhs_concat_dim); in GenNewConvDNums()
267 input_spatial_dimensions.push_back(dot_lhs->shape().dimensions_size() - 1); in GenNewConvDNums()
324 for (auto dim : input_spatial_dimensions) { in GenNewConvDNums()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dattribute_importer.cc91 dnums.input_spatial_dimensions().begin(), in ConvertConvDimensionNumbers()
92 dnums.input_spatial_dimensions().end()); in ConvertConvDimensionNumbers()
102 arrayref(dnums.input_spatial_dimensions()), in ConvertConvDimensionNumbers()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dpooling.cc178 std::vector<int64_t> input_spatial_dimensions; in MakeSpatialPadding() local
185 input_spatial_dimensions.push_back(input_size[dim]); in MakeSpatialPadding()
189 return MakePadding(input_spatial_dimensions, kernel_size_spatial_dimensions, in MakeSpatialPadding()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/python/
H A Dattributes.py97 input_spatial_dimensions=[1, 2, 3],
108 assert attr.input_spatial_dimensions == [1, 2, 3]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/
H A Dverifier_conv_op.mlir14 input_spatial_dimensions = [1, 2],
41 input_spatial_dimensions = [1, 2],
199 input_spatial_dimensions = [1, 4],
227 input_spatial_dimensions = [1, 2],
255 input_spatial_dimensions = [1, 2],
514 input_spatial_dimensions = [1, 2],
542 input_spatial_dimensions = [1, 2],
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/tests/
H A Dverify_conv.mlir14 input_spatial_dimensions = [1, 2],
41 input_spatial_dimensions = [1, 2],
199 input_spatial_dimensions = [1, 4],
227 input_spatial_dimensions = [1, 2],
255 input_spatial_dimensions = [1, 2],
514 input_spatial_dimensions = [1, 2],
542 input_spatial_dimensions = [1, 2],
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dconv_canonicalization.cc67 new_input_dim_order[i + 1] = dnums.input_spatial_dimensions(i); in Run()
69 input->shape().dimensions(dnums.input_spatial_dimensions(i)); in Run()
H A Dir_emission_utils.cc91 if (dnums.input_spatial_dimensions(i) != i + 1) { in PotentiallyImplementedAsEigenConvolution()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_client.py516 self.input_spatial_dimensions = []
564 dimension_numbers.input_spatial_dimensions.extend(range(2, 2 + nd))
580 dimension_numbers.input_spatial_dimensions.extend(
H A Dxla_client.pyi142 input_spatial_dimensions: List[int]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc857 auto input_spatial_dimensions = dnums.getInputSpatialDimensions(); in IsSupportedConvOp() local
858 for (auto p : llvm::enumerate(input_spatial_dimensions)) { in IsSupportedConvOp()
916 auto input_spatial_dimensions = dnums.getInputSpatialDimensions(); in MatchResizeOp() local
919 if (input_spatial_dimensions.size() != 2 || in MatchResizeOp()
922 input_spatial_dimensions[0] != output_spatial_dimensions[0] || in MatchResizeOp()
923 input_spatial_dimensions[1] != output_spatial_dimensions[1]) in MatchResizeOp()
950 static_cast<int>(lhs_type.getDimSize(input_spatial_dimensions[0])), in MatchResizeOp()
951 static_cast<int>(lhs_type.getDimSize(input_spatial_dimensions[1]))}; in MatchResizeOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dlayouts_and_names.mlir13 input_spatial_dimensions = [1, 2],
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util.cc433 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(0)); in ConvArray4DGeneralDimensionsDilated()
435 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(1)); in ConvArray4DGeneralDimensionsDilated()

12