summaryrefslogtreecommitdiffstats
path: root/tests/dnn/dnn-layer-conv2d-test.c
diff options
context:
space:
mode:
authorGuo, Yejun <yejun.guo@intel.com>2019-10-21 20:38:03 +0800
committerPedro Arthur <bygrandao@gmail.com>2019-10-30 10:31:55 -0300
commitdff39ea9f0154ec52b7548b122a4a5332df3c2c6 (patch)
treea435c8d24d80bc9c38ecd4b93c51113c47acad05 /tests/dnn/dnn-layer-conv2d-test.c
parenta269fa044b1364af1654456c33b7d45407822876 (diff)
downloadffmpeg-streaming-dff39ea9f0154ec52b7548b122a4a5332df3c2c6.zip
ffmpeg-streaming-dff39ea9f0154ec52b7548b122a4a5332df3c2c6.tar.gz
dnn: add tf.nn.conv2d support for native model
Unlike other tf.*.conv2d layers, tf.nn.conv2d does not create many nodes (within a scope) in the graph, it just acts like other layers. tf.nn.conv2d only creates one node in the graph, and no internal nodes such as 'kernel' are created. The format of native model file is also changed, a flag named has_bias is added, so change the version number. Signed-off-by: Guo, Yejun <yejun.guo@intel.com> Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'tests/dnn/dnn-layer-conv2d-test.c')
-rw-r--r--tests/dnn/dnn-layer-conv2d-test.c2
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/dnn/dnn-layer-conv2d-test.c b/tests/dnn/dnn-layer-conv2d-test.c
index 9d13da3..2da01e5 100644
--- a/tests/dnn/dnn-layer-conv2d-test.c
+++ b/tests/dnn/dnn-layer-conv2d-test.c
@@ -97,6 +97,7 @@ static int test_with_same_dilate(void)
float bias[2] = { -1.6574852, -0.72915393 };
params.activation = TANH;
+ params.has_bias = 1;
params.biases = bias;
params.dilation = 2;
params.input_num = 3;
@@ -196,6 +197,7 @@ static int test_with_valid(void)
float bias[2] = { -0.4773722, -0.19620377 };
params.activation = TANH;
+ params.has_bias = 1;
params.biases = bias;
params.dilation = 1;
params.input_num = 3;
OpenPOWER on IntegriCloud