[XLA:CPU] Add support for MKLDNN contraction kernel in runtime_matmul.cc

PiperOrigin-RevId: 225538587

tensorflow/compiler/xla/service/cpu/BUILD

@@ -572,6 +572,7 @@ cc_library(
         ":runtime_matvec",
         "//tensorflow/compiler/xla:executable_run_options",
         "//tensorflow/core:framework_lite",
+        "//tensorflow/core/kernels:eigen_contraction_kernel",
         "//third_party/eigen3",
     ],
 )

tensorflow/compiler/xla/service/cpu/runtime_matmul.cc

@@ -23,6 +23,10 @@ limitations under the License.
 #include "tensorflow/core/platform/dynamic_annotations.h"
 #include "tensorflow/core/platform/types.h"
 
+#if defined(TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL)
+#include "tensorflow/core/kernels/eigen_contraction_kernel.h"
+#endif
+
 using tensorflow::int32;
 using tensorflow::int64;