Return status correctly.

Author: mdfaijul

tensorflow/core/kernels/mkl/mkl_kernel_util.h

@@ -55,7 +55,7 @@ class MklTestingUtil {
   // and max values to be in float type, min_tensor and max_tensor should have
   // their dtype set to DT_FLOAT.
   template <typename T>
-  static void GetQuantizationTensors(const Tensor& input, Tensor* output,
+  static Status GetQuantizationTensors(const Tensor& input, Tensor* output,
                                      DataType out_type, const string mode,
                                      Tensor* min_tensor, Tensor* max_tensor) {
     if (min_tensor->dtype() != DT_FLOAT || max_tensor->dtype() != DT_FLOAT) {

tensorflow/core/kernels/mkl/onednn_fused_matmul_ops_test.cc

@@ -346,9 +346,10 @@ class FusedMatMulOpsTest : public OpsTestBase {
     Tensor x_qtensor(qinput_dtype, x.shape());
     Tensor x_min_tensor(DT_FLOAT, TensorShape({}));
     Tensor x_max_tensor(DT_FLOAT, TensorShape({}));
-    MklTestingUtil::GetQuantizationTensors<T>(x, &x_qtensor, qinput_dtype,
-                                              input_quant_mode, &x_min_tensor,
-                                              &x_max_tensor);
+    auto status = MklTestingUtil::GetQuantizationTensors<T>(
+        x, &x_qtensor, qinput_dtype, input_quant_mode, &x_min_tensor,
+        &x_max_tensor);
+    ASSERT_TRUE(status.ok());
     Tensor y_qtensor(DT_QINT8, y.shape());
     const int num_channels = transpose_y ? y.dim_size(0) : y.dim_size(1);
     TensorShape minmax_shape =
@@ -364,8 +365,9 @@ class FusedMatMulOpsTest : public OpsTestBase {
                                                 &y_max_tensor);
       }
     } else {
-      MklTestingUtil::GetQuantizationTensors<T>(
+      auto status = MklTestingUtil::GetQuantizationTensors<T>(
           y, &y_qtensor, DT_QINT8, "SCALED", &y_min_tensor, &y_max_tensor);
+      ASSERT_TRUE(status.ok());
     }
 
     Scope root = tensorflow::Scope::NewRootScope();