Function TFDataFormatToMklDnnDataFormat

tensorflow/core/util/mkl_util.h:1038–1043

Map TensorFlow data format into OneDNN data format @input: TensorFlow data format @return: OneDNN data format corresponding to TensorFlow data format; Fails with an error if invalid data format.

/// @return: OneDNN data format corresponding to TensorFlow data format;
///          Fails with an error if invalid data format.
inline MklTensorFormat TFDataFormatToMklDnnDataFormat(TensorFormat format) {
  if (format == FORMAT_NHWC) return MklTensorFormat::FORMAT_NHWC;
  if (format == FORMAT_NCHW) return MklTensorFormat::FORMAT_NCHW;
  TF_CHECK_OK(Status(error::Code::INVALID_ARGUMENT, "Unsupported data format"));
  return MklTensorFormat::FORMAT_INVALID;
}

/// Map OneDNN data format into TensorFlow data format
///