message: Adding support for 2 inputs in MKL Dequatize op
diff: @@ -84,7 +84,7 @@ class MklDequantizeOp : public OpKernel {
  auto src_md = src_mkl_shape.IsMklTensor()
  ? src_mkl_shape.GetMklLayout()
  : memory::desc(src_dims, MklDnnType<T>(),
  - memory::format_tag::nhwc);
  + src_dims.size() == 4 ? MEMORY_FORMAT::nhwc : MEMORY_FORMAT::nc);
  src.SetUsrMem(src_md, &src_tensor);
  src.SetUsrMemDataHandle(&src_tensor, reorder_stream);
  @@ -100,7 +100,7 @@ class MklDequantizeOp : public OpKernel {
  dst_md.data.data_type = memory::convert_to_c(MklDnnType<float>());
  } else {
  dst_md = memory::desc(src_dims, MklDnnType<float>(),
  - memory::format_tag::nhwc);
  + src_dims.size() == 4 ? MEMORY_FORMAT::nhwc : MEMORY_FORMAT::nc);
  }
  // If input is MKL shape, output is also MKL shape.
  