| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_input_ops.cc | 255 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size,
256 dims.spatial_dims[0].stride, padding_,
257 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
261 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size,
262 dims.spatial_dims[1].stride, padding_,
263 &dims.spatial_dims[1].output_size, &pad_left, &pad_right));
269 out_backprop.tensor<T, 4>(), dims.spatial_dims[0].input_size,
270 dims.spatial_dims[1].input_size
[all...] |
| conv_grad_filter_ops.cc | 252 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size,
253 dims.spatial_dims[0].stride, padding_,
254 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
258 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size,
259 dims.spatial_dims[1].stride, padding_,
260 &dims.spatial_dims[1].output_size, &pad_left, &pad_right));
266 dims.spatial_dims[0].input_size, dims.spatial_dims[1].input_size
[all...] |
| mkl_conv_grad_filter_ops.cc | 140 backprop_dims.spatial_dims[0].input_size,
141 backprop_dims.spatial_dims[0].filter_size,
142 backprop_dims.spatial_dims[0].stride, padding_,
143 &backprop_dims.spatial_dims[0].output_size,
146 backprop_dims.spatial_dims[1].input_size,
147 backprop_dims.spatial_dims[1].filter_size,
148 backprop_dims.spatial_dims[1].stride, padding_,
149 &backprop_dims.spatial_dims[1].output_size,
160 static_cast<size_t>(backprop_dims.spatial_dims[1].input_size);
162 static_cast<size_t>(backprop_dims.spatial_dims[0].input_size)
[all...] |
| mkl_conv_grad_input_ops.cc | 145 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size,
146 dims.spatial_dims[0].stride, padding,
147 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
151 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size,
152 dims.spatial_dims[1].stride, padding,
153 &dims.spatial_dims[1].output_size, &pad_left, &pad_right));
158 static_cast<size_t>(dims.spatial_dims[1].input_size);
160 static_cast<size_t>(dims.spatial_dims[0].input_size)
[all...] |
| conv_grad_ops.h | 230 gtl::InlinedVector<ConvBackpropSpatialDimension, 3> spatial_dims; member in struct:tensorflow::ConvBackpropDimensions
|
| conv_grad_ops.cc | 140 dims->spatial_dims.resize(num_spatial_dims);
145 strides, padding, image_dim, i, &dims->spatial_dims[i]));
|
| /external/tensorflow/tensorflow/python/ops/ |
| nn_ops.py | 215 spatial_dims=None,
238 adjusted_dilation_rate is an int64 tensor of shape [max(spatial_dims)],
239 adjusted_{paddings,crops} are int64 tensors of shape [max(spatial_dims), 2]
250 input_shape[spatial_dims],
259 input_shape[spatial_dims],
266 `spatial_dims` may not be, we must adjust `dilation_rate`, `paddings` and
270 which is what is needed for dimensions not part of `spatial_dims`.
274 For 0 <= i < len(spatial_dims), we assign:
276 adjusted_dilation_rate[spatial_dims[i] - 1] = dilation_rate[i]
277 adjusted_paddings[spatial_dims[i] - 1, :] = paddings[i, :
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| conv_ops.cc | 421 padding[i] = {dims.spatial_dims[i].pad_before,
422 dims.spatial_dims[i].pad_after};
423 lhs_dilation[i] = dims.spatial_dims[i].stride;
594 dims.spatial_dims[i].expanded_output_size +
595 (dims.spatial_dims[i].filter_size - 1) * dilations_[dim];
611 const int64 pad_total = padded_in_size - dims.spatial_dims[i].input_size;
625 rhs_dilation[i] = dims.spatial_dims[i].stride;
|
| pooling_ops.cc | 481 padding->set_edge_padding_low(dims.spatial_dims[i].pad_before);
482 padding->set_edge_padding_high(dims.spatial_dims[i].pad_after);
483 padding->set_interior_padding(dims.spatial_dims[i].stride - 1);
|
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.cc | 317 gtl::MutableArraySlice<DimensionHandle> spatial_dims,
320 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format);
324 for (int spatial_dim_index = 0; spatial_dim_index < spatial_dims.size();
326 spatial_dims[spatial_dim_index] = context->Dim(
341 gtl::ArraySlice<DimensionHandle> spatial_dims,
344 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format);
350 for (int spatial_dim_index = 0; spatial_dim_index < spatial_dims.size();
353 rank, format, spatial_dim_index)] = spatial_dims[spatial_dim_index];
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | 316 const std::vector<int64>& spatial_dims);
605 const std::vector<int64>& spatial_dims) {
610 for (int dim : spatial_dims) {
618 for (int dim : spatial_dims) {
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 543 const auto& spatial_dims = filter_descriptor.input_filter_dims(); local
544 std::copy(spatial_dims.begin(), spatial_dims.end(), dims.begin() + 2);
[all...] |
