add safety checks for mkldnn-assignement for quantized inner product (#2815)

* - add safety checks for mkldnn-assignement for quantized inner product

* - add asserts for unsupported data types in builder & emitter code of Quantized Dot

src/ngraph/runtime/cpu/builder/quantized_dot.cpp

@@ -36,6 +36,12 @@ namespace ngraph
             {
                 if (runtime::cpu::mkldnn_utils::use_mkldnn_kernel(node))
                 {
+                    if (node->get_input_element_type(0) == element::u8 &&
+                        node->get_input_element_type(1) == element::u8)
+                    {
+                        throw ngraph_error(
+                            "Unsupported data types for QuantizedDot MKLDNN kernel.");
+                    }
                     auto& functors = external_function->get_functors();
                     auto& arg0_tensor = external_function->get_tensor_data(args[0].get_name());
                     auto& arg1_tensor = external_function->get_tensor_data(args[1].get_name());

src/ngraph/runtime/cpu/cpu_emitter.cpp

@@ -2494,6 +2494,13 @@ namespace ngraph
             {
                 if (runtime::cpu::mkldnn_utils::use_mkldnn_kernel(node))
                 {
+                    if (node->get_input_element_type(0) == element::u8 &&
+                        node->get_input_element_type(1) == element::u8)
+                    {
+                        throw ngraph_error(
+                            "Unsupported data types for QuantizedDot MKLDNN kernel.");
+                    }
+
                     auto& mkldnn_emitter = external_function->get_mkldnn_emitter();
                     auto qip_index = mkldnn_emitter->build_inner_product<ngraph::op::QuantizedDot>(
                         node, args, out);

src/ngraph/runtime/cpu/pass/cpu_assignment.cpp

@@ -782,13 +782,21 @@ namespace ngraph
                 template <>
                 void CPUAssignment::ASSIGN_DECL(ngraph::op::QuantizedDotBias)
                 {
-                    runtime::cpu::mkldnn_utils::assign_mkldnn_kernel(node);
+                    if (node->get_input_element_type(0) == element::u8 &&
+                        node->get_input_element_type(1) == element::i8)
+                    {
+                        runtime::cpu::mkldnn_utils::assign_mkldnn_kernel(node);
+                    }
                 }
 
                 template <>
                 void CPUAssignment::ASSIGN_DECL(ngraph::op::QuantizedDot)
                 {
-                    runtime::cpu::mkldnn_utils::assign_mkldnn_kernel(node);
+                    if (node->get_input_element_type(0) == element::u8 &&
+                        node->get_input_element_type(1) == element::i8)
+                    {
+                        runtime::cpu::mkldnn_utils::assign_mkldnn_kernel(node);
+                    }
                 }
 
                 template <>