Home
last modified time | relevance | path

Searched refs:output_spatial_dimensions (Results 1 – 25 of 47) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_vectorize_convolutions_test.cc88 ASSERT_EQ(dnums.output_spatial_dimensions().size(), 2); in TEST_F()
101 EXPECT_EQ(dnums.output_spatial_dimensions()[0], 1); in TEST_F()
102 EXPECT_EQ(dnums.output_spatial_dimensions()[1], 2); in TEST_F()
161 ASSERT_EQ(dnums.output_spatial_dimensions().size(), 2); in TEST_F()
175 EXPECT_EQ(dnums.output_spatial_dimensions()[0], 3); in TEST_F()
176 EXPECT_EQ(dnums.output_spatial_dimensions()[1], 4); in TEST_F()
214 ASSERT_EQ(dnums.output_spatial_dimensions().size(), 2); in TEST_F()
228 EXPECT_EQ(dnums.output_spatial_dimensions()[0], 3); in TEST_F()
229 EXPECT_EQ(dnums.output_spatial_dimensions()[1], 4); in TEST_F()
267 ASSERT_EQ(dnums.output_spatial_dimensions().size(), 2); in TEST_F()
[all …]
H A Dstream_executor_util.cc138 dnums.output_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
139 dnums.output_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
146 dnums.output_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
147 dnums.output_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
153 dnums.output_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
154 dnums.output_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
308 dnums.output_feature_dimension(), dnums.output_spatial_dimensions()), in FindVectorizedFeatureDims()
H A Dgpu_conv_rewriter.cc110 auto output_spatial_dims = conv_dnums.output_spatial_dimensions(); in MatchBackwardFilter()
347 const auto& output_spatial_dims = dnums.output_spatial_dimensions(); in MatchBackwardInput()
459 conv_dnums.output_spatial_dimensions(i)); in MatchBackwardInput()
H A Dcudnn_support_utils.cc72 dnums.output_spatial_dimensions().size() != 2) { in CudnnSupportsOptimizedIntegerConvolution()
H A Dcudnn_simplify_padding.cc64 for (int64_t d : dnums.output_spatial_dimensions()) { in FindOutputVectCDim()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtranspose_folding_test.cc408 dnums.output_spatial_dimensions(0), in TEST_F()
409 new_conv->convolution_dimension_numbers().output_spatial_dimensions(0)); in TEST_F()
411 dnums.output_spatial_dimensions(1), in TEST_F()
412 new_conv->convolution_dimension_numbers().output_spatial_dimensions(1)); in TEST_F()
477 dnums.output_spatial_dimensions(0), in TEST_F()
478 new_conv->convolution_dimension_numbers().output_spatial_dimensions(0)); in TEST_F()
480 dnums.output_spatial_dimensions(1), in TEST_F()
481 new_conv->convolution_dimension_numbers().output_spatial_dimensions(1)); in TEST_F()
H A Dconvolution_4d_expander.cc76 int64_t output_spatial_dim = dim_nums.output_spatial_dimensions(i); in ExpandInstruction()
141 for (int64_t i = 0; i < new_dim_nums.output_spatial_dimensions().size(); in ExpandInstruction()
145 new_dim_nums.output_spatial_dimensions(i))); in ExpandInstruction()
H A Dspace_to_batch_converter.cc383 const int64_t output_spatial_dim = dim_numbers.output_spatial_dimensions( in IsForwardWindowDilatedConv()
2598 permuted_conv_dims_numbers.output_spatial_dimensions( in PropagateOnConv()
2689 dim_translator[permuted_conv_dims_numbers.output_spatial_dimensions(j)] = in PropagateOnConv()
2731 original_conv_dims.output_spatial_dimensions( in PropagateOnConv()
3161 permuted_conv_dims_numbers.output_spatial_dimensions( in PropagateOnBackpropFilterConv()
3433 new_dim_numbers.output_spatial_dimensions( in PropagateOnBackpropFilterConv()
3447 original_conv_dims.output_spatial_dimensions( in PropagateOnBackpropFilterConv()
3654 const int64_t output_spatial_dim = dim_numbers.output_spatial_dimensions( in PerformSpaceToBatchOnConvolution()
3762 dim_translator[dim_numbers.output_spatial_dimensions(j)] = dim_count; in PerformSpaceToBatchOnConvolution()
3806 dim_numbers.output_spatial_dimensions(first_dim + i); in PerformSpaceToBatchOnConvolution()
[all …]
H A Dhlo_cse.cc189 absl::MakeSpan(conv_dimension_numbers.output_spatial_dimensions())); in AbslHashValue()
H A Ddot_as_convolution_util.cc99 int64_t output = conv_dims.output_spatial_dimensions(i); in ParseConvolutionDimsInfo()
H A Dshape_inference.cc1698 std::copy(dnums.output_spatial_dimensions().begin(), in InferConvolveShape()
1699 dnums.output_spatial_dimensions().end(), output_dnums.begin() + 2); in InferConvolveShape()
1826 dimensions[dnums.output_spatial_dimensions(i)] = in InferConvolveShape()
1842 is_dynamic[dnums.output_spatial_dimensions(j)] = true; in InferConvolveShape()
1861 is_dynamic[dnums.output_spatial_dimensions(j)] = true; in InferConvolveShape()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dattribute_importer.cc97 dnums.output_spatial_dimensions().begin(), in ConvertConvDimensionNumbers()
98 dnums.output_spatial_dimensions().end()); in ConvertConvDimensionNumbers()
107 arrayref(dnums.output_spatial_dimensions())); in ConvertConvDimensionNumbers()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/python/
H A Dattributes.py103 output_spatial_dimensions=[0, 2])
114 assert attr.output_spatial_dimensions == [0, 2]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/
H A Dverifier_conv_op.mlir20 output_spatial_dimensions = [1, 2]
47 output_spatial_dimensions = [1, 2]
205 output_spatial_dimensions = [1, 2]
233 output_spatial_dimensions = [1, 2]
261 output_spatial_dimensions = [0, 3]
520 output_spatial_dimensions = [1, 2]
548 output_spatial_dimensions = [1, 2]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/tests/
H A Dverify_conv.mlir20 output_spatial_dimensions = [1, 2]
47 output_spatial_dimensions = [1, 2]
205 output_spatial_dimensions = [1, 2]
233 output_spatial_dimensions = [1, 2]
261 output_spatial_dimensions = [0, 3]
520 output_spatial_dimensions = [1, 2]
548 output_spatial_dimensions = [1, 2]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dconv_canonicalization.cc110 new_output_dim_order[i + 1] = dnums.output_spatial_dimensions(i); in Run()
112 hlo->shape().dimensions(dnums.output_spatial_dimensions(i)); in Run()
H A Dir_emission_utils.cc97 if (dnums.output_spatial_dimensions(i) != i + 1) { in PotentiallyImplementedAsEigenConvolution()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_client.py522 self.output_spatial_dimensions = []
566 dimension_numbers.output_spatial_dimensions.extend(range(2, 2 + nd))
583 dimension_numbers.output_spatial_dimensions.extend(
H A Dxla_client.pyi148 output_spatial_dimensions: List[int]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc870 auto output_spatial_dimensions = dnums.getOutputSpatialDimensions(); in IsSupportedConvOp() local
871 for (auto p : llvm::enumerate(output_spatial_dimensions)) { in IsSupportedConvOp()
918 auto output_spatial_dimensions = dnums.getOutputSpatialDimensions(); in MatchResizeOp() local
920 output_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()
953 output_spatial_dimensions[0])), in MatchResizeOp()
955 output_spatial_dimensions[1]))}; in MatchResizeOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc91 dnums.output_spatial_dimensions(i); in PartitionConvolutionWithBatchGroupCount()
182 dnums.output_spatial_dimensions(i); in PartitionConvolutionWithFeatureGroupCount()
766 dnums.output_spatial_dimensions(i); in PartitionConvolutionTiledOutput()
H A Ddot_handler.cc298 std::vector<int64_t> output_spatial_dimensions( in GenNewConvDNums() local
299 dnums.output_spatial_dimensions().begin(), in GenNewConvDNums()
300 dnums.output_spatial_dimensions().end()); in GenNewConvDNums()
311 for (int64_t i = 0; i < output_spatial_dimensions.size(); ++i) { in GenNewConvDNums()
312 if (output_slice_dim <= output_spatial_dimensions[i]) { in GenNewConvDNums()
313 output_spatial_dimensions[i]++; in GenNewConvDNums()
316 output_spatial_dimensions.push_back(output_slice_dim); in GenNewConvDNums()
318 output_spatial_dimensions.push_back(new_dot_shape.dimensions_size() - 1); in GenNewConvDNums()
335 for (auto dim : output_spatial_dimensions) { in GenNewConvDNums()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dlayouts_and_names.mlir19 output_spatial_dimensions = [2, 3]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/lhlo_gpu/
H A Dlhlo_gpu_ops.mlir23 output_spatial_dimensions = [2, 3]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dxla_ops_test.py146 dnums.output_spatial_dimensions.extend(range(2, 2 + num_spatial_dims))
184 dnums.output_spatial_dimensions.extend(range(2, 2 + num_spatial_dims))

12