| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 399 dnums.output_spatial_dimensions(0), in TEST_F()
400 new_conv->convolution_dimension_numbers().output_spatial_dimensions(0)); in TEST_F()
402 dnums.output_spatial_dimensions(1), in TEST_F()
403 new_conv->convolution_dimension_numbers().output_spatial_dimensions(1)); in TEST_F()
467 dnums.output_spatial_dimensions(0), in TEST_F()
468 new_conv->convolution_dimension_numbers().output_spatial_dimensions(0)); in TEST_F()
470 dnums.output_spatial_dimensions(1), in TEST_F()
471 new_conv->convolution_dimension_numbers().output_spatial_dimensions(1)); in TEST_F()
|
| D | convolution_4d_expander.cc | 76 int64 output_spatial_dim = dim_nums.output_spatial_dimensions(i); in ExpandInstruction()
140 for (int64 i = 0; i < new_dim_nums.output_spatial_dimensions().size(); ++i) { in ExpandInstruction()
143 new_dim_nums.output_spatial_dimensions(i))); in ExpandInstruction()
|
| D | space_to_batch_converter.cc | 1910 permuted_conv_dims_numbers.output_spatial_dimensions( in PropagateOnConv()
1994 dim_map[permuted_conv_dims_numbers.output_spatial_dimensions(j)] = in PropagateOnConv()
2028 original_conv_dims.output_spatial_dimensions( in PropagateOnConv()
2305 permuted_conv_dims_numbers.output_spatial_dimensions( in PropagateOnBackpropFilterConv()
2544 new_dim_numbers.output_spatial_dimensions( in PropagateOnBackpropFilterConv()
2554 original_conv_dims.output_spatial_dimensions( in PropagateOnBackpropFilterConv()
2674 const int64 output_spatial_dim = dim_numbers.output_spatial_dimensions( in PerformSpaceToBatchOnConvolution()
2757 dim_map[dim_numbers.output_spatial_dimensions(j)] = dim_count; in PerformSpaceToBatchOnConvolution()
2794 new_dim_numbers.output_spatial_dimensions( in PerformSpaceToBatchOnConvolution()
2807 dim_numbers.output_spatial_dimensions( in PerformSpaceToBatchOnConvolution()
[all …]
|
| D | dot_as_convolution_util.cc | 72 int64 output = conv_dims.output_spatial_dimensions(i); in ParseConvolutionDimsInfo()
|
| D | shape_inference.cc | 1699 std::copy(dnums.output_spatial_dimensions().begin(), in InferConvolveShape()
1700 dnums.output_spatial_dimensions().end(), output_dnums.begin() + 2); in InferConvolveShape()
1825 dimensions[dnums.output_spatial_dimensions(i)] = in InferConvolveShape()
1841 is_dynamic[dnums.output_spatial_dimensions(j)] = true; in InferConvolveShape()
1860 is_dynamic[dnums.output_spatial_dimensions(j)] = true; in InferConvolveShape()
|
| D | sharding_propagation.cc | 137 instruction->shape().dimensions(dnums.output_spatial_dimensions(i)); in IsConvolutionKernelSmall()
527 output_to_lhs_indices[dnums.output_spatial_dimensions(i)] = in InferConvolutionShardingFromOperands()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo_gpu_ops.mlir | 45 output_spatial_dimensions = dense<[2,3]> : tensor<2xi64>},
80 output_spatial_dimensions = dense<[1, 2]> : tensor<2xi64>},
110 output_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>},
142 output_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>},
173 output_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>},
|
| D | lhlo-legalize-to-linalg.mlir | 908 …_batch_dimension = 0 : i64, output_feature_dimension = 3 : i64, output_spatial_dimensions = dense<…
915 …_batch_dimension = 0 : i64, output_feature_dimension = 3 : i64, output_spatial_dimensions = dense<…
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.cc | 102 dnums.output_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
103 dnums.output_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
108 dnums.output_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
109 dnums.output_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
|
| D | gpu_conv_rewriter.cc | 189 auto output_spatial_dims = conv_dnums.output_spatial_dimensions(); in MatchBackwardFilter()
399 const auto& output_spatial_dims = dnums.output_spatial_dimensions(); in MatchBackwardInput()
511 conv_dnums.output_spatial_dimensions(i)); in MatchBackwardInput()
|
| D | gpu_conv_runner.cc | 425 output_shape.dimensions(dnums.output_spatial_dimensions(dim))); in GetGpuConvConfig()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization.cc | 108 new_output_dim_order[i + 1] = dnums.output_spatial_dimensions(i); in Run()
110 hlo->shape().dimensions(dnums.output_spatial_dimensions(i)); in Run()
|
| D | ir_emission_utils.cc | 99 if (dnums.output_spatial_dimensions(i) != i + 1) { in PotentiallyImplementedAsEigenConvolution()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 113 dnums.output_spatial_dimensions().begin(), in ConvertConvDimensionNumbers()
114 dnums.output_spatial_dimensions().end()); in ConvertConvDimensionNumbers()
|
| D | attribute_exporter.cc | 53 for (auto v : input.output_spatial_dimensions().getValues<int64>()) { in ConvertConvDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | layouts_and_names.mlir | 19 output_spatial_dimensions = dense<[ 2, 3 ]> : tensor<2xi64>
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 547 self.output_spatial_dimensions = []
591 dimension_numbers.output_spatial_dimensions.extend(range(2, 2 + nd))
608 dimension_numbers.output_spatial_dimensions.extend(
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base_structs.td | 50 StructFieldAttr<"output_spatial_dimensions", I64ElementsAttr>] > {
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 283 std::vector<int64> output_spatial_dimensions( in GenNewConvDNums() local
284 dnums.output_spatial_dimensions().begin(), in GenNewConvDNums()
285 dnums.output_spatial_dimensions().end()); in GenNewConvDNums()
296 for (int64 i = 0; i < output_spatial_dimensions.size(); ++i) { in GenNewConvDNums()
297 if (output_slice_dim <= output_spatial_dimensions[i]) { in GenNewConvDNums()
298 output_spatial_dimensions[i]++; in GenNewConvDNums()
301 output_spatial_dimensions.push_back(output_slice_dim); in GenNewConvDNums()
303 output_spatial_dimensions.push_back(new_dot_shape.dimensions_size() - 1); in GenNewConvDNums()
320 for (auto dim : output_spatial_dimensions) { in GenNewConvDNums()
|
| D | convolution_handler.cc | 91 dnums.output_spatial_dimensions(i); in PartitionConvolutionWithBatchGroupCount()
182 dnums.output_spatial_dimensions(i); in PartitionConvolutionWithFeatureGroupCount()
764 dnums.output_spatial_dimensions(i); in PartitionConvolutionTiledOutput()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 292 // CHECK-SAME: output_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>
329 // CHECK-SAME: output_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>
368 // CHECK-SAME: output_spatial_dimensions = dense<[2, 3]> : tensor<2xi64>
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 241 DenseIntElementsAttr output_spatial_dimensions = in IsSupportedConvOp() local
242 conv_op.dimension_numbers().output_spatial_dimensions(); in IsSupportedConvOp()
244 llvm::enumerate(output_spatial_dimensions.getValues<int64_t>())) { in IsSupportedConvOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 547 dim_nums.output_feature_dimension(), dim_nums.output_spatial_dimensions(), in EmitConvolutionForwardAsMlir()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 146 dnums.output_spatial_dimensions.extend(range(2, 2 + num_spatial_dims))
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 294 llvm::size(dimension_numbers.output_spatial_dimensions()); in matchAndRewrite()
311 dimension_numbers.output_spatial_dimensions().begin(); in matchAndRewrite()
|