Adding filter rank to SVDF
Bug: 67597181
Pulled in the latest implementation from TF Lite.
This allows a rank parameter that is greater than 1.
Split the state into input and output states.
Added a test spec for rank 2 SVDF.
Test: NeuralNetworksTest
Test: SVDFTest: adb shell /data/nativetest64/svdf_test/svdf_test
Change-Id: I207cc081019369bf5e643e16e91ff664d10975cc
(cherry picked from commit 062414c5c5e20d7432d17b00eb7daf0562396815)
diff --git a/common/Android.bp b/common/Android.bp
index 1e251cc..109d133 100644
--- a/common/Android.bp
+++ b/common/Android.bp
@@ -163,6 +163,9 @@
"operations/SVDFTest.cpp",
],
local_include_dirs: [ "include" ],
+ header_libs: [
+ "tensorflow_headers",
+ ],
shared_libs: [
"libneuralnetworks",
],