experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request from PariksheetPinjari909:patch-8

Browse Source 
PiperOrigin-RevId: 261397325
This commit is contained in:
TensorFlower Gardener 2019-08-02 16:16:02 -07:00
commit a4f3f64132
1 changed files with 6 additions and 0 deletions

6
tensorflow/core/util/tensor_format.h
View File

@ -124,6 +124,9 @@ inline int GetTensorSpatialDims(int num_dims, TensorFormat format) {
// Note: the VECT_W is not counted as an independent spatial dim here, // Note: the VECT_W is not counted as an independent spatial dim here,
// since it just a component of the width dimension. // since it just a component of the width dimension.
return num_dims - 3; // Exclude N,C,VectDim. return num_dims - 3; // Exclude N,C,VectDim.
default:
LOG(FATAL) << "Unknown format " << format;
return -1; // Avoid compiler warning about missing return value
} }
} }
@ -148,6 +151,9 @@ inline int GetTensorDimsFromSpatialDims(int num_spatial_dims,
case FORMAT_NCHW_VECT_C: case FORMAT_NCHW_VECT_C:
case FORMAT_NHWC_VECT_W: case FORMAT_NHWC_VECT_W:
return num_spatial_dims + 3; // Include N,C,VectDim. return num_spatial_dims + 3; // Include N,C,VectDim.
default:
LOG(FATAL) << "Unknown format " << format;
return -1; // Avoid compiler warning about missing return value
} }
} }