Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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()
139 ASSERT_EQ(dnums.input_spatial_dimensions().size(), 2); in TEST_F()
145 EXPECT_EQ(dnums.input_spatial_dimensions()[0], 3); in TEST_F()
146 EXPECT_EQ(dnums.input_spatial_dimensions()[1], 4); in TEST_F()
192 ASSERT_EQ(dnums.input_spatial_dimensions().size(), 2); in TEST_F()
198 EXPECT_EQ(dnums.input_spatial_dimensions()[0], 3); in TEST_F()
199 EXPECT_EQ(dnums.input_spatial_dimensions()[1], 4); in TEST_F()
245 ASSERT_EQ(dnums.input_spatial_dimensions().size(), 2); in TEST_F()
[all …]
Dgpu_conv_padding_legalization.cc64 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) { in MaybePaddedAndSlicedInput()
65 int64_t dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput()
98 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) { in MaybePaddedAndSlicedInput()
99 int64_t dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput()
236 int64_t dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardFilterConvolution()
320 int64_t dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()
364 int64_t dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()
Dstream_executor_util.cc77 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
78 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
85 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
86 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
92 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
93 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
301 dnums.input_spatial_dimensions()), in FindVectorizedFeatureDims()
Dgpu_conv_rewriter.cc183 auto input_spatial_dims = conv_dnums.input_spatial_dimensions(); in MatchBackwardFilter()
425 const auto& input_spatial_dims = dnums.input_spatial_dimensions(); in MatchBackwardInput()
540 conv_dnums.input_spatial_dimensions(i)); in MatchBackwardInput()
/external/tensorflow/tensorflow/compiler/xla/service/
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()
131 for (int64_t i = 0; i < new_dim_nums.input_spatial_dimensions().size(); ++i) { in ExpandInstruction()
134 new_dim_nums.input_spatial_dimensions(i))); in ExpandInstruction()
Dtranspose_folding_test.cc393 dnums.input_spatial_dimensions(0), in TEST_F()
394 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0)); in TEST_F()
396 dnums.input_spatial_dimensions(1), in TEST_F()
397 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1)); in TEST_F()
461 dnums.input_spatial_dimensions(0), in TEST_F()
462 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1)); in TEST_F()
464 dnums.input_spatial_dimensions(1), in TEST_F()
465 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0)); in TEST_F()
Ddot_as_convolution_util.cc94 int64_t lhs = conv_dims.input_spatial_dimensions(i); in ParseConvolutionDimsInfo()
144 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
154 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
Dspace_to_batch_converter.cc606 auto it = absl::c_find(dim_numbers.input_spatial_dimensions(), i); in BringSpaceNextToBatch()
607 if (it != dim_numbers.input_spatial_dimensions().end()) { in BringSpaceNextToBatch()
608 int64_t j = it - dim_numbers.input_spatial_dimensions().begin(); in BringSpaceNextToBatch()
938 if (dim_numbers.input_spatial_dimensions( in CanPropagate()
942 for (int i = 0; i < dim_numbers.input_spatial_dimensions().size(); in CanPropagate()
944 if (dim_numbers.input_spatial_dimensions(i) == dim_map.batch || in CanPropagate()
945 dim_numbers.input_spatial_dimensions(i) == dim_map.feature) { in CanPropagate()
2214 const int64_t old_space_dim = original_conv_dims.input_spatial_dimensions( in PropagateOnConv()
2233 original_conv_dims.input_spatial_dimensions(i))); in PropagateOnConv()
2548 int64_t old_space_dim = original_conv_dims.input_spatial_dimensions( in PropagateOnBackpropFilterConv()
[all …]
Ddynamic_padder.cc921 dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad()
947 [&](int64_t dim) { return dnums.input_spatial_dimensions(dim); }); in RewriteDynamicConvolutionInputGrad()
990 dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionForward()
1017 [&](int64_t dim) { return dnums.input_spatial_dimensions(dim); }); in RewriteDynamicConvolutionForward()
1052 dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionKernelGrad()
1102 [&](int64_t dim) { return dnums.input_spatial_dimensions(dim); }); in RewriteDynamicConvolutionKernelGrad()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
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()
323 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
360 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
537 dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
631 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
659 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
[all …]
Ddot_handler.cc151 if (conv_dnums.input_spatial_dimensions(i) == lhs_concat_dim) { in GenNewWindow()
239 std::vector<int64> input_spatial_dimensions( in GenNewConvDNums() local
240 dnums.input_spatial_dimensions().begin(), in GenNewConvDNums()
241 dnums.input_spatial_dimensions().end()); in GenNewConvDNums()
249 for (int64_t i = 0; i < input_spatial_dimensions.size(); ++i) { in GenNewConvDNums()
250 if (lhs_concat_dim <= input_spatial_dimensions[i]) { in GenNewConvDNums()
251 input_spatial_dimensions[i]++; in GenNewConvDNums()
254 input_spatial_dimensions.push_back(lhs_concat_dim); in GenNewConvDNums()
258 input_spatial_dimensions.push_back(dot_lhs->shape().dimensions_size() - 1); in GenNewConvDNums()
315 for (auto dim : input_spatial_dimensions) { in GenNewConvDNums()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops_base_structs.cc82 print_dim(dnums.input_spatial_dimensions(), in printConvolutionDimensions()
193 DenseIntElementsAttr input_spatial_dimensions = parsed_dims.first; in parseConvolutionDimensions() local
213 input_batch_dimension, input_feature_dimension, input_spatial_dimensions, in parseConvolutionDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc65 new_input_dim_order[i + 1] = dnums.input_spatial_dimensions(i); in Run()
67 input->shape().dimensions(dnums.input_spatial_dimensions(i)); in Run()
Dir_emission_utils.cc93 if (dnums.input_spatial_dimensions(i) != i + 1) { in PotentiallyImplementedAsEigenConvolution()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc178 std::vector<int64> 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()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc107 dnums.input_spatial_dimensions().begin(), in ConvertConvDimensionNumbers()
108 dnums.input_spatial_dimensions().end()); in ConvertConvDimensionNumbers()
Dattribute_exporter.cc35 for (auto v : input.input_spatial_dimensions().getValues<int64>()) { in ConvertConvDimensionNumbers()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dlayouts_and_names.mlir13 input_spatial_dimensions = dense<[ 1, 2 ]> : tensor<2xi64>,
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py565 self.input_spatial_dimensions = []
613 dimension_numbers.input_spatial_dimensions.extend(range(2, 2 + nd))
629 dimension_numbers.input_spatial_dimensions.extend(
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc112 conv_op.dimension_numbers().input_spatial_dimensions().getNumElements(); in matchAndRewrite()
150 .input_spatial_dimensions() in IsSamePadding()
252 conv_op.dimension_numbers().input_spatial_dimensions(), in CreateConvOp()
366 conv_op.dimension_numbers().input_spatial_dimensions().getNumElements(); in IsSupportedConvOp()
414 .input_spatial_dimensions() in matchAndRewrite()
505 conv_op.dimension_numbers().input_spatial_dimensions().getNumElements(); in IsSupportedConvOp()
518 DenseIntElementsAttr input_spatial_dimensions = in IsSupportedConvOp() local
519 conv_op.dimension_numbers().input_spatial_dimensions(); in IsSupportedConvOp()
521 llvm::enumerate(input_spatial_dimensions.getValues<int64_t>())) { in IsSupportedConvOp()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base_structs.td44 StructFieldAttr<"input_spatial_dimensions", I64ElementsAttr>,
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc431 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(0)); in ConvArray4DGeneralDimensionsDilated()
433 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(1)); in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir44 input_spatial_dimensions = dense<[2,3]> : tensor<2xi64>,
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc539 dim_nums.input_spatial_dimensions(), builder); in EmitConvolutionForwardAsMlir()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py145 dnums.input_spatial_dimensions.extend(range(2, 2 + num_spatial_dims))
183 dnums.input_spatial_dimensions.extend(range(2, 2 + num_spatial_dims))

12