Method IsMklChannelDim

tensorflow/core/util/mkl_util.h:532–534

Query TF-OneDNN dimension ordering map and check if Tensorflow dimension 'd' corresponds to OneDNN's Channel dimension.

  /// Query TF-OneDNN dimension ordering map and check if Tensorflow dimension 'd'
  /// corresponds to OneDNN's Channel dimension.
  inline bool IsMklChannelDim(int d) const {
    return TfDimIdx(d) == MklDnnDims::Dim_C;
  }

  /// Query TF-OneDNN dimension ordering map and check if Tensorflow dimension 'd'
  /// corresponds to OneDNN's Batch dimension.