Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_4d_expander.cc41 if (dim_nums.input_spatial_dimensions().size() != 4) { in InstructionMatchesPattern()
45 for (int64 i = 0; i < dim_nums.input_spatial_dimensions().size(); ++i) { in InstructionMatchesPattern()
46 int64 spatial_dim = dim_nums.input_spatial_dimensions(i); in InstructionMatchesPattern()
74 for (int64 i = 0; i < dim_nums.input_spatial_dimensions().size(); ++i) { in ExpandInstruction()
75 int64 input_spatial_dim = dim_nums.input_spatial_dimensions(i); in ExpandInstruction()
131 for (int64 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.cc68 int64 lhs = conv_dims.input_spatial_dimensions(i); in ParseConvolutionDimsInfo()
118 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
128 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
Dspace_to_batch_converter.cc540 auto it = absl::c_find(dim_numbers.input_spatial_dimensions(), i); in BringSpaceNextToBatch()
541 if (it != dim_numbers.input_spatial_dimensions().end()) { in BringSpaceNextToBatch()
542 int64 j = it - dim_numbers.input_spatial_dimensions().begin(); in BringSpaceNextToBatch()
766 if (consumer->convolution_dimension_numbers().input_spatial_dimensions( in CanPropagate()
1854 const int64 old_space_dim = original_conv_dims.input_spatial_dimensions( in PropagateOnConv()
1873 original_conv_dims.input_spatial_dimensions(i))); in PropagateOnConv()
2162 int64 old_space_dim = original_conv_dims.input_spatial_dimensions( in PropagateOnBackpropFilterConv()
2290 original_conv_dims.input_spatial_dimensions(i))); in PropagateOnBackpropFilterConv()
2325 permuted_conv_dims_numbers.input_spatial_dimensions( in PropagateOnBackpropFilterConv()
2602 dim_numbers.input_spatial_dimensions(get_chosen_spatial_dim(convolution)); in GetConvolutionDetails()
Ddynamic_padder.cc822 int64 input_spatial_dim = dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad()
848 [&](int64 dim) { return dnums.input_spatial_dimensions(dim); }); in RewriteDynamicConvolutionInputGrad()
890 int64 input_spatial_dim = dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionForward()
917 [&](int64 dim) { return dnums.input_spatial_dimensions(dim); }); in RewriteDynamicConvolutionForward()
951 int64 input_spatial_dim = dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionKernelGrad()
1001 [&](int64 dim) { return dnums.input_spatial_dimensions(dim); }); in RewriteDynamicConvolutionKernelGrad()
Dshape_inference.cc1685 std::copy(dnums.input_spatial_dimensions().begin(), in InferConvolveShape()
1686 dnums.input_spatial_dimensions().end(), input_dnums.begin() + 2); in InferConvolveShape()
1736 input_spatial_dims[i] = lhs.dimensions(dnums.input_spatial_dimensions(i)); in InferConvolveShape()
1838 if (i == dnums.input_spatial_dimensions(j)) { in InferConvolveShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc64 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) { in MaybePaddedAndSlicedInput()
65 int64 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 dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput()
236 int64 dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardFilterConvolution()
320 int64 dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()
364 int64 dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()
Dstream_executor_util.cc58 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
59 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
64 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
65 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
Dgpu_conv_rewriter.cc183 auto input_spatial_dims = conv_dnums.input_spatial_dimensions(); in MatchBackwardFilter()
398 const auto& input_spatial_dims = dnums.input_spatial_dimensions(); in MatchBackwardInput()
513 conv_dnums.input_spatial_dimensions(i)); in MatchBackwardInput()
Dgpu_conv_runner.cc384 input_shape.dimensions(dnums.input_spatial_dimensions(dim))); in GetGpuConvConfig()
/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 lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
360 int64 lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
537 dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
631 int64 lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
659 int64 lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
[all …]
Ddot_handler.cc147 if (conv_dnums.input_spatial_dimensions(i) == lhs_concat_dim) { in GenNewWindow()
234 std::vector<int64> input_spatial_dimensions( in GenNewConvDNums() local
235 dnums.input_spatial_dimensions().begin(), in GenNewConvDNums()
236 dnums.input_spatial_dimensions().end()); in GenNewConvDNums()
244 for (int64 i = 0; i < input_spatial_dimensions.size(); ++i) { in GenNewConvDNums()
245 if (lhs_concat_dim <= input_spatial_dimensions[i]) { in GenNewConvDNums()
246 input_spatial_dimensions[i]++; in GenNewConvDNums()
249 input_spatial_dimensions.push_back(lhs_concat_dim); in GenNewConvDNums()
253 input_spatial_dimensions.push_back(dot_lhs->shape().dimensions_size() - 1); in GenNewConvDNums()
309 for (auto dim : input_spatial_dimensions) { in GenNewConvDNums()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir39 input_spatial_dimensions = dense<[2,3]> : tensor<2xi64>,
74 input_spatial_dimensions = dense<[1, 2]> : tensor<2xi64>,
104 input_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>,
136 input_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>,
167 input_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>,
/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.py541 self.input_spatial_dimensions = []
589 dimension_numbers.input_spatial_dimensions.extend(range(2, 2 + nd))
605 dimension_numbers.input_spatial_dimensions.extend(
/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/mlir/tensorflow/transforms/
Dlegalize_hlo.cc121 .input_spatial_dimensions() in matchAndRewrite()
219 conv_op.dimension_numbers().input_spatial_dimensions().getNumElements(); in IsSupportedConvOp()
229 DenseIntElementsAttr input_spatial_dimensions = in IsSupportedConvOp() local
230 conv_op.dimension_numbers().input_spatial_dimensions(); in IsSupportedConvOp()
232 llvm::enumerate(input_spatial_dimensions.getValues<int64_t>())) { in IsSupportedConvOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt286 // CHECK-SAME: input_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>
323 // CHECK-SAME: input_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>
362 // CHECK-SAME: input_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc430 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(0)); in ConvArray4DGeneralDimensionsDilated()
432 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(1)); in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc538 dim_nums.input_spatial_dimensions(), builder); in EmitConvolutionForwardAsMlir()

12