Searched refs:spatial_dim (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 216 int spatial_dim) { in GetTensorSpatialDimIndex() argument
217 CHECK(spatial_dim >= 0 && in GetTensorSpatialDimIndex()
218 spatial_dim < GetTensorSpatialDims(num_dims, format)) in GetTensorSpatialDimIndex()
219 << spatial_dim << " " << num_dims << " " << ToString(format); in GetTensorSpatialDimIndex()
223 return spatial_dim + 1; in GetTensorSpatialDimIndex()
226 return spatial_dim + 2; in GetTensorSpatialDimIndex()
229 return spatial_dim; in GetTensorSpatialDimIndex()
590 for (int spatial_dim = 0; spatial_dim < num_src_spatial_dims; ++spatial_dim) { in ShapeFromFormat() local
591 spatial_dims[spatial_dim] = in ShapeFromFormat()
593 src_shape.dims(), src_format, spatial_dim)]; in ShapeFromFormat()
|
| D | tensor_format_test.cc | 50 int spatial(int spatial_index) const { return spatial_dim[spatial_index]; } in spatial()
53 int spatial_dim[3]; member
62 int spatial(int spatial_index) const { return spatial_dim[spatial_index]; } in spatial()
65 int spatial_dim[3]; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops.cc | 61 Padding padding, int64 padding_before, int64 padding_after, int spatial_dim, in ConvBackpropExtractAndVerifyDimension() argument
63 dim->input_size = input_shape.dim_size(spatial_dim); in ConvBackpropExtractAndVerifyDimension()
65 dim->output_size = output_shape.dim_size(spatial_dim); in ConvBackpropExtractAndVerifyDimension()
66 dim->stride = strides[spatial_dim]; in ConvBackpropExtractAndVerifyDimension()
67 dim->dilation = dilations[spatial_dim]; in ConvBackpropExtractAndVerifyDimension()
76 " spatial_dim: ", spatial_dim, " input: ", dim->input_size, in ConvBackpropExtractAndVerifyDimension()
|
| D | 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/ |
| D | pooling_ops.cc | 141 for (int spatial_dim = 0; spatial_dim < num_spatial_dims; ++spatial_dim) { in XlaTensorFormat() local
142 spatial_dimensions[spatial_dim] = in XlaTensorFormat()
143 GetTensorSpatialDimIndex(num_dims, data_format, spatial_dim); in XlaTensorFormat()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 344 for (int spatial_dim = 0; spatial_dim < spatial.size(); spatial_dim++) { in MakeShapeFromFormat() local
345 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] = in MakeShapeFromFormat()
346 context->MakeDim(spatial[spatial_dim]); in MakeShapeFromFormat()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 262 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); }
|