Function MklDnnDimsToTFShape

tensorflow/core/util/mkl_util.h:1162–1171

Map OneDNN memory::dims object into TensorShape object. This function will simply map input shape in OneDNN memory::dims format in Tensorflow's TensorShape object by preserving dimension order. @input OneDNN memory::dims object @output TensorShape corresponding to memory::dims

/// @input OneDNN memory::dims object
/// @output TensorShape corresponding to memory::dims
inline TensorShape MklDnnDimsToTFShape(const memory::dims& dims) {
  std::vector<int32> shape(dims.size(), -1);
  for (int d = 0; d < dims.size(); d++) {
    shape[d] = dims[d];
  }

  TensorShape ret;
  CHECK_EQ(TensorShapeUtils::MakeShape(shape, &ret).ok(), true);
  return ret;
}

/// Function to calculate strides given tensor shape in Tensorflow order
/// E.g., if dims_tf_order is {1, 2, 3, 4}, then as per Tensorflow convention,