Function TFDataFormatToMklDnn3DDataFormat

tensorflow/core/util/mkl_util.h:1026–1031

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

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

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