| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_input_ops.cc | 351 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size,
352 dims.spatial_dims[0].stride, padding_,
353 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
357 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size,
358 dims.spatial_dims[1].stride, padding_,
359 &dims.spatial_dims[1].output_size, &pad_left, &pad_right));
365 out_backprop.tensor<T, 4>(), dims.spatial_dims[0].input_size,
366 dims.spatial_dims[1].input_size
[all...] |
| conv_grad_ops.h | 236 gtl::InlinedVector<ConvBackpropSpatialDimension, 3> spatial_dims; member in struct:tensorflow::ConvBackpropDimensions
245 int64 input_size(int dim) const { return spatial_dims[dim].input_size; }
246 int64 filter_size(int dim) const { return spatial_dims[dim].filter_size; }
247 int64 output_size(int dim) const { return spatial_dims[dim].output_size; }
248 int64 stride(int dim) const { return spatial_dims[dim].stride; }
249 int64 dilation(int dim) const { return spatial_dims[dim].dilation; }
|
| conv_grad_filter_ops.cc | 270 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size,
271 dims.spatial_dims[0].stride, padding_,
272 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
276 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size,
277 dims.spatial_dims[1].stride, padding_,
278 &dims.spatial_dims[1].output_size, &pad_left, &pad_right));
285 dims.spatial_dims[0].input_size, dims.spatial_dims[1].input_size
[all...] |
| conv_grad_ops_3d.cc | 247 static_cast<int>(dims.spatial_dims[0].stride), // stride_planes
248 static_cast<int>(dims.spatial_dims[1].stride), // stride_rows
249 static_cast<int>(dims.spatial_dims[2].stride)); // stride_cols
353 dims.spatial_dims[0].input_size,
354 dims.spatial_dims[0].filter_size,
355 dims.spatial_dims[0].stride, padding_,
356 &dims.spatial_dims[0].output_size,
359 dims.spatial_dims[1].input_size,
360 dims.spatial_dims[1].filter_size,
361 dims.spatial_dims[1].stride, padding_
[all...] |
| conv_grad_ops.cc | 143 dims->spatial_dims.resize(num_spatial_dims);
154 &dims->spatial_dims[i]));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| conv_op_helpers.cc | 374 padding[i] = {dims.spatial_dims[i].pad_before,
375 dims.spatial_dims[i].pad_after};
376 lhs_dilation[i] = dims.spatial_dims[i].stride;
477 rhs_dilation[i] = dims.spatial_dims[i].stride;
486 dims.spatial_dims[i].expanded_output_size +
487 (dims.spatial_dims[i].filter_size - 1) * attrs.dilations[dim];
503 const int64 pad_total = padded_in_size - dims.spatial_dims[i].input_size;
|
| /external/tensorflow/tensorflow/python/ops/ |
| nn_ops.py | 328 spatial_dims=None,
351 adjusted_dilation_rate is an int64 tensor of shape [max(spatial_dims)],
352 adjusted_{paddings,crops} are int64 tensors of shape [max(spatial_dims), 2]
363 input_shape[spatial_dims],
372 input_shape[spatial_dims],
379 `spatial_dims` may not be, we must adjust `dilation_rate`, `paddings` and
383 which is what is needed for dimensions not part of `spatial_dims`.
387 For 0 <= i < len(spatial_dims), we assign:
389 adjusted_dilation_rate[spatial_dims[i] - 1] = dilation_rate[i]
390 adjusted_paddings[spatial_dims[i] - 1, :] = paddings[i, :
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| tensor_format.h | 589 std::vector<int64> spatial_dims(num_src_spatial_dims);
591 spatial_dims[spatial_dim] =
596 spatial_dims[num_src_spatial_dims - 1] *= 4;
598 return ShapeFromFormat(dst_format, batch, {spatial_dims}, channels);
|
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.cc | 354 gtl::MutableArraySlice<DimensionHandle> spatial_dims,
357 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format);
361 for (int spatial_dim_index = 0; spatial_dim_index < spatial_dims.size();
363 spatial_dims[spatial_dim_index] = context->Dim(
378 gtl::ArraySlice<DimensionHandle> spatial_dims,
381 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format);
387 for (int spatial_dim_index = 0; spatial_dim_index < spatial_dims.size();
390 rank, format, spatial_dim_index)] = spatial_dims[spatial_dim_index];
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | 341 const std::vector<int64>& spatial_dims);
676 const std::vector<int64>& spatial_dims) {
681 for (int dim : spatial_dims) {
689 for (int dim : spatial_dims) {
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 572 absl::Span<const int64> spatial_dims = local
574 std::copy(spatial_dims.begin(), spatial_dims.end(), dims.begin() + 2);
[all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.cc | 614 const auto& spatial_dims = filter_descriptor.input_filter_dims(); local
615 std::copy(spatial_dims.begin(), spatial_dims.end(), dims.begin() + 2);
[all...] |
