Function TFShapeToMklDnnDims

tensorflow/core/util/mkl_util.h:1072–1078

Map TensorShape object into memory::dims required by OneDNN This function will simply map input TensorShape into OneDNN dims naively. So it will preserve the order of dimensions. E.g., if input tensor is in NHWC format, then dims will be in NHWC format also. @input TensorShape object in shape @return memory::dims corresponding to TensorShape

/// @input TensorShape object in shape
/// @return memory::dims corresponding to TensorShape
inline memory::dims TFShapeToMklDnnDims(const TensorShape& shape) {
  memory::dims dims(shape.dims());
  for (int d = 0; d < shape.dims(); ++d) {
    dims[d] = shape.dim_size(d);
  }
  return dims;
}

/// Map TensorShape object into memory::dims in NCHW format required by OneDNN
///