Function IsMklOp

tensorflow/core/graph/mkl_graph_util.h:248–250

Check if the operator with 'op_name' and type 'T' is an OneDNN operator that will either understand input tensors in OneDNN layout or will go through name rewrite that some operators go through.

// will either understand input tensors in OneDNN layout or will go through name
// rewrite that some operators go through.
static inline bool IsMklOp(const string& op_name, DataType T) {
  return IsMklLayoutDependentOp(op_name, T) || IsMklNameChangeOp(op_name, T);
}

static inline bool IsMklOp(const Node* n) {
  DataType T;