| /external/tensorflow/tensorflow/core/util/ |
| tensor_format.h | 211 // Returns the dimension index of the specified 'spatial_dim' within an
212 // activation tensor. If format is NHWC_VECT_W and spatial_dim is 1, returns
216 int spatial_dim) {
217 CHECK(spatial_dim >= 0 &&
218 spatial_dim < GetTensorSpatialDims(num_dims, format))
219 << spatial_dim << " " << num_dims << " " << ToString(format);
223 return spatial_dim + 1;
226 return spatial_dim + 2;
229 return spatial_dim;
590 for (int spatial_dim = 0; spatial_dim < num_src_spatial_dims; ++spatial_dim)
[all...] |
| tensor_format_test.cc | 50 int spatial(int spatial_index) const { return spatial_dim[spatial_index]; }
53 int spatial_dim[3]; member in struct:tensorflow::TensorDimMap
62 int spatial(int spatial_index) const { return spatial_dim[spatial_index]; }
65 int spatial_dim[3]; member in struct:tensorflow::FilterDimMap
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_ops.cc | 61 Padding padding, int64 padding_before, int64 padding_after, int spatial_dim,
63 dim->input_size = input_shape.dim_size(spatial_dim);
65 dim->output_size = output_shape.dim_size(spatial_dim);
66 dim->stride = strides[spatial_dim];
67 dim->dilation = dilations[spatial_dim];
76 " spatial_dim: ", spatial_dim, " input: ", dim->input_size,
|
| conv_2d.h | 177 for (int spatial_dim = 0; spatial_dim < NDIMS - 2; ++spatial_dim) {
178 expanded_dims[out_index++] = in.dimension(spatial_dim);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| pooling_ops.cc | 141 for (int spatial_dim = 0; spatial_dim < num_spatial_dims; ++spatial_dim) {
142 spatial_dimensions[spatial_dim] =
143 GetTensorSpatialDimIndex(num_dims, data_format, spatial_dim);
|
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.cc | 344 for (int spatial_dim = 0; spatial_dim < spatial.size(); spatial_dim++) {
345 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] =
346 context->MakeDim(spatial[spatial_dim]);
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | 262 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } function in class:stream_executor::dnn::BatchDescriptor
[all...] |
