Returns the index of the inner width dimension.
| 214 | |
| 215 | // Returns the index of the inner width dimension. |
| 216 | inline int GetTensorInnerWidthDimIndex(int num_dims, TensorFormat format) { |
| 217 | DCHECK_EQ(format, FORMAT_NHWC_VECT_W); |
| 218 | return num_dims - 1; |
| 219 | } |
| 220 | |
| 221 | // Returns the dimension index of the specified 'spatial_dim' within an |
| 222 | // activation tensor. If format is NHWC_VECT_W and spatial_dim is 1, returns |
no outgoing calls
no test coverage detected