| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 142 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size, in operator ()()
145 DCHECK_EQ(dims.spatial_dims[0].output_size, expected_out_rows); in operator ()()
147 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size, in operator ()()
150 DCHECK_EQ(dims.spatial_dims[1].output_size, expected_out_cols); in operator ()()
469 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size, in Compute()
470 dims.spatial_dims[0].stride, padding_, in Compute()
471 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom)); in Compute()
475 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size, in Compute()
476 dims.spatial_dims[1].stride, padding_, in Compute()
477 &dims.spatial_dims[1].output_size, &pad_left, &pad_right)); in Compute()
[all …]
|
| D | conv_grad_input_ops.h | 150 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size, in operator()
153 DCHECK_EQ(dims.spatial_dims[0].output_size, expected_out_rows); in operator()
156 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size, in operator()
159 DCHECK_EQ(dims.spatial_dims[1].output_size, expected_out_cols); in operator()
563 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size,
564 dims.spatial_dims[0].stride, padding_,
565 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
569 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size,
570 dims.spatial_dims[1].stride, padding_,
571 &dims.spatial_dims[1].output_size, &pad_left, &pad_right));
[all …]
|
| D | conv_grad_input_ops.cc | 106 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size, in operator ()()
109 DCHECK_EQ(dims.spatial_dims[0].output_size, expected_out_rows); in operator ()()
111 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size, in operator ()()
114 DCHECK_EQ(dims.spatial_dims[1].output_size, expected_out_cols); in operator ()()
131 if (dims.spatial_dims[0].filter_size == 1 && in operator ()()
132 dims.spatial_dims[1].filter_size == 1 && !is_grouped_convolution && in operator ()()
133 dims.spatial_dims[0].stride == 1 && dims.spatial_dims[1].stride == 1 && in operator ()()
136 const uint64 m = dims.batch_size * dims.spatial_dims[0].input_size * in operator ()()
137 dims.spatial_dims[1].input_size; in operator ()()
161 } else if (dims.spatial_dims[0].filter_size == in operator ()()
[all …]
|
| D | conv_grad_ops_3d.cc | 257 static_cast<int>(dims.spatial_dims[0].stride), // stride_planes in Compute()
258 static_cast<int>(dims.spatial_dims[1].stride), // stride_rows in Compute()
259 static_cast<int>(dims.spatial_dims[2].stride)); // stride_cols in Compute()
364 dims.spatial_dims[0].input_size, in Compute()
365 dims.spatial_dims[0].filter_size, in Compute()
366 dims.spatial_dims[0].stride, padding_, in Compute()
367 &dims.spatial_dims[0].output_size, in Compute()
370 dims.spatial_dims[1].input_size, in Compute()
371 dims.spatial_dims[1].filter_size, in Compute()
372 dims.spatial_dims[1].stride, padding_, in Compute()
[all …]
|
| D | conv_grad_shape_utils.h | 47 gtl::InlinedVector<ConvBackpropSpatialDimension, 3> spatial_dims; member
56 int64 input_size(int dim) const { return spatial_dims[dim].input_size; } in input_size()
57 int64 filter_size(int dim) const { return spatial_dims[dim].filter_size; } in filter_size()
58 int64 output_size(int dim) const { return spatial_dims[dim].output_size; } in output_size()
59 int64 stride(int dim) const { return spatial_dims[dim].stride; } in stride()
60 int64 dilation(int dim) const { return spatial_dims[dim].dilation; } in dilation()
|
| D | conv_2d.h | 212 Eigen::DSizes<IndexType, NDIMS - 2> spatial_dims;
213 for (int i = 0; i < spatial_dims.rank(); ++i) {
214 spatial_dims[i] = in.dimension(i);
219 merged_dims[0] = spatial_dims.TotalSize(); // product of spatial dims [H*W]
233 for (int i = 0; i < spatial_dims.rank(); ++i) {
234 expanded_dims[2 + i] = spatial_dims[i];
242 for (int i = 0; i < spatial_dims.rank(); ++i) {
243 expanded_dims[1 + i] = spatial_dims[i];
|
| D | conv_grad_shape_utils.cc | 139 dims->spatial_dims.resize(num_spatial_dims); in ConvBackpropComputeDimensionsV2()
150 &dims->spatial_dims[i])); in ConvBackpropComputeDimensionsV2()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_to_space_nd_test.cc | 74 int spatial_dims = static_cast<int>(block_shape.size()); in BatchToSpaceNDOpConstModel() local
76 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {spatial_dims}); in BatchToSpaceNDOpConstModel()
77 crops_ = AddConstInput(TensorType_INT32, crops, {spatial_dims, 2}); in BatchToSpaceNDOpConstModel()
104 int spatial_dims = static_cast<int>(input_shape.size()) - 2; in BatchToSpaceNDOpDynamicModel() local
108 BuildInterpreter({input_shape, {spatial_dims}, {spatial_dims, 2}}); in BatchToSpaceNDOpDynamicModel()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 421 spatial_dims=None, argument
569 spatial_dims=spatial_dims,
599 spatial_dims=None, argument
620 if spatial_dims is None:
621 spatial_dims = range(starting_spatial_dim,
623 orig_spatial_dims = list(spatial_dims)
624 spatial_dims = sorted(set(int(x) for x in orig_spatial_dims))
625 if spatial_dims != orig_spatial_dims or any(x < 1 for x in spatial_dims):
631 expected_input_rank = spatial_dims[-1]
633 expected_input_rank = spatial_dims[-1] + 1
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 361 padding[i] = {dims.spatial_dims[i].pad_before, in MakeXlaBackpropInputConvOp()
362 dims.spatial_dims[i].pad_after}; in MakeXlaBackpropInputConvOp()
363 lhs_dilation[i] = dims.spatial_dims[i].stride; in MakeXlaBackpropInputConvOp()
474 rhs_dilation[i] = dims.spatial_dims[i].stride; in MakeXlaBackpropFilterConvOp()
483 dims.spatial_dims[i].expanded_output_size + in MakeXlaBackpropFilterConvOp()
484 (dims.spatial_dims[i].filter_size - 1) * attrs.dilations[dim]; in MakeXlaBackpropFilterConvOp()
500 const int64 pad_total = padded_in_size - dims.spatial_dims[i].input_size; in MakeXlaBackpropFilterConvOp()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 602 std::vector<int64> spatial_dims(num_src_spatial_dims); in ShapeFromFormat()
604 spatial_dims[spatial_dim] = in ShapeFromFormat()
609 spatial_dims[num_src_spatial_dims - 1] *= 4; in ShapeFromFormat()
611 return ShapeFromFormat(dst_format, batch, {spatial_dims}, channels); in ShapeFromFormat()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 68 absl::Span<const tensorflow::protobuf_int64> spatial_dims, in GetShapeInfo() argument
80 for (int64 dim : spatial_dims) { in GetShapeInfo()
100 /*dimCount=*/2 + spatial_dims.size(), /*symbolCount=*/0, affine_exprs, in GetShapeInfo()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 521 gtl::MutableArraySlice<DimensionHandle> spatial_dims, in DimensionsFromShape() argument
524 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in DimensionsFromShape()
528 for (int spatial_dim_index = 0, end = spatial_dims.size(); in DimensionsFromShape()
530 spatial_dims[spatial_dim_index] = context->Dim( in DimensionsFromShape()
545 gtl::ArraySlice<DimensionHandle> spatial_dims, in ShapeFromDimensions() argument
548 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in ShapeFromDimensions()
554 for (int spatial_dim_index = 0, end = spatial_dims.size(); in ShapeFromDimensions()
557 rank, format, spatial_dim_index)] = spatial_dims[spatial_dim_index]; in ShapeFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1100 DenseIntElementsAttr spatial_dims = in GetConvDimensionNumbersAttr() local
1115 batch_dim, feature_dim, spatial_dims, kernel_input_feature_dim, in GetConvDimensionNumbersAttr()
1117 feature_dim, spatial_dims, builder->getContext())); in GetConvDimensionNumbersAttr()
4166 SmallVector<int64_t, num_spatial_dims> spatial_dims; in matchAndRewrite() local
4173 spatial_dims.push_back(dim); in matchAndRewrite()
4174 const auto &spatial_dim_i = dims.spatial_dims[i]; in matchAndRewrite()
4185 auto spatial_dims_attr = GetI64ElementsAttr(spatial_dims, &rewriter); in matchAndRewrite()
4361 SmallVector<int64_t, num_spatial_dims> spatial_dims; in matchAndRewrite() local
4383 const auto &spatial_dim_i = dims.spatial_dims[i]; in matchAndRewrite()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 345 …'input\', \'dilation_rate\', \'padding\', \'op\', \'filter_shape\', \'spatial_dims\', \'data_forma…
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 342 const std::vector<int64>& spatial_dims);
677 const std::vector<int64>& spatial_dims) { in ImageDims() argument
682 for (int dim : spatial_dims) { in ImageDims()
690 for (int dim : spatial_dims) { in ImageDims()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 433 …'input\', \'dilation_rate\', \'padding\', \'op\', \'filter_shape\', \'spatial_dims\', \'data_forma…
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 594 absl::Span<const int64> spatial_dims = in CudnnFilterDescriptor() local
596 std::copy(spatial_dims.begin(), spatial_dims.end(), dims.begin() + 2); in CudnnFilterDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 725 const auto& spatial_dims = filter_descriptor.input_filter_dims(); in ScopedFilterDescriptor() local
726 std::copy(spatial_dims.begin(), spatial_dims.end(), dims.begin() + 2); in ScopedFilterDescriptor()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 756 * `spatial_dims`: Describes the `n` spatial dimensions that define the base
765 * `spatial_dims`: Describes the `n` spatial dimensions that define the n-d
827 * `spatial_dims`: One value for each valid placement of the convolutional
|