/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...] |