diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index b07e018dd2a555..6b3df3ea4101be 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -1586,6 +1586,7 @@ cc_library( "//tensorflow/core/kernels:quantized_ops", "//tensorflow/core/kernels/neon:neon_depthwise_conv_op", ]) + if_mkl([ + "//tensorflow/core/kernels:mkl_aggregate_ops", "//tensorflow/core/kernels:mkl_concat_op", "//tensorflow/core/kernels:mkl_dequantize_op", "//tensorflow/core/kernels:mkl_conv_op", @@ -1594,8 +1595,9 @@ cc_library( "//tensorflow/core/kernels:mkl_identity_op", "//tensorflow/core/kernels:mkl_input_conversion_op", "//tensorflow/core/kernels:mkl_lrn_op", - "//tensorflow/core/kernels:mkl_requantize_ops", "//tensorflow/core/kernels:mkl_pooling_ops", + "//tensorflow/core/kernels:mkl_qmatmul_op", + "//tensorflow/core/kernels:mkl_requantize_ops", "//tensorflow/core/kernels:mkl_quantize_op", "//tensorflow/core/kernels:mkl_relu_op", "//tensorflow/core/kernels:mkl_reshape_op", @@ -1603,7 +1605,6 @@ cc_library( "//tensorflow/core/kernels:mkl_softmax_op", "//tensorflow/core/kernels:mkl_transpose_op", "//tensorflow/core/kernels:mkl_tfconv_op", - "//tensorflow/core/kernels:mkl_aggregate_ops", ]) + if_cuda([ "//tensorflow/core/grappler/optimizers:gpu_swapping_kernels", "//tensorflow/core/grappler/optimizers:gpu_swapping_ops", @@ -4414,6 +4415,7 @@ tf_cc_test_mkl( "//tensorflow/core/kernels:mkl_input_conversion_op", "//tensorflow/core/kernels:mkl_lrn_op", "//tensorflow/core/kernels:mkl_pooling_ops", + "//tensorflow/core/kernels:mkl_qmatmul_op", "//tensorflow/core/kernels:mkl_quantize_op", "//tensorflow/core/kernels:mkl_relu_op", "//tensorflow/core/kernels:mkl_reshape_op", diff --git a/tensorflow/core/api_def/base_api/api_def_QuantizedMatMulWithBias.pbtxt b/tensorflow/core/api_def/base_api/api_def_QuantizedMatMulWithBias.pbtxt new file mode 100644 index 00000000000000..140144961f3c87 --- /dev/null +++ b/tensorflow/core/api_def/base_api/api_def_QuantizedMatMulWithBias.pbtxt @@ -0,0 +1,84 @@ +op { + graph_op_name: "QuantizedMatMulWithBias" + visibility: HIDDEN + in_arg { + name: "a" + description: <