[INTEL oneDNN][Bug Fix] Fix incorrect use of int for dim size related to oneDNN gemm and matmul op
This fixes incorrect use of int for dim size when invoking oneDNN gemm and matmul primitive. 
This causes overflow here:
     https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/mkl/mkl_matmul_ops_common.h#L52

and thus results in some large matmuls running single thread.

