diff options
author | Guo, Yejun <yejun.guo@intel.com> | 2019-10-21 20:38:03 +0800 |
---|---|---|
committer | Pedro Arthur <bygrandao@gmail.com> | 2019-10-30 10:31:55 -0300 |
commit | dff39ea9f0154ec52b7548b122a4a5332df3c2c6 (patch) | |
tree | a435c8d24d80bc9c38ecd4b93c51113c47acad05 /tests/dnn/dnn-layer-conv2d-test.c | |
parent | a269fa044b1364af1654456c33b7d45407822876 (diff) | |
download | ffmpeg-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.c | 2 |
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; |