commit 8c8c15947c9e80824fdf736cec220556d1207f9f
author Marat Dukhan <maratek@google.com> Tue Jul 13 13:59:02 2021 -0700
committer XNNPACK Team <xnnpack-github-robot@google.com> Tue Jul 13 14:00:10 2021 -0700
tree b290485a89b5b0d0f77703f14d2ddbb79d935671
parent ac67ae890df607035031cd1a570b1fcc24e2ee54

Expose QU8 [Depthwise] Convolution 2D operators in Subgraph API

PiperOrigin-RevId: 384546022

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index 772eb3d..e10646c 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -5724,7 +5724,6 @@
         "//conditions:default": [],
     }),
     defines = [
-        "XNN_NO_QU8_OPERATORS",
         "XNN_NO_U8_OPERATORS",
         "XNN_NO_X8_OPERATORS",
         "XNN_NO_F16_OPERATORS",
@@ -5739,6 +5738,14 @@
             "XNN_NO_QC8_OPERATORS",
             "XNN_NO_QS8_OPERATORS",
         ],
+    }) + select({
+        ":xnn_enable_qu8_explicit_true": [],
+        ":xnn_enable_qu8_explicit_false": [
+            "XNN_NO_QU8_OPERATORS",
+        ],
+        "//conditions:default": [
+            "XNN_NO_QU8_OPERATORS",
+        ],
     }),
     gcc_copts = xnnpack_gcc_std_copts(),
     includes = ["include"],
@@ -8425,6 +8432,18 @@
     define_values = {"xnn_enable_qs8": "false"},
 )
 
+# Enable QU8 inference in TFLite-specific version
+config_setting(
+    name = "xnn_enable_qu8_explicit_true",
+    define_values = {"xnn_enable_qu8": "true"},
+)
+
+# Disable QU8 inference in TFLite-specific version
+config_setting(
+    name = "xnn_enable_qu8_explicit_false",
+    define_values = {"xnn_enable_qu8": "false"},
+)
+
 # Builds with -c dbg
 config_setting(
     name = "debug_build",