Add custom contraction kernel for QInt8xQInt8->QInt32 contraction.

PiperOrigin-RevId: 243678946
tensorflow · Apr 15, 2019 · dfcbe0a · dfcbe0a
1 parent b40daed
commit dfcbe0a
Show file tree

Hide file tree

Showing 3 changed files with 346 additions and 172 deletions.
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD
@@ -2730,6 +2730,7 @@ tf_cc_test(
         ":eigen_contraction_kernel",
         "//tensorflow/core:test",
         "//tensorflow/core:test_main",
+        "//third_party/eigen3",
     ],
 )