diff options
author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-03-13 11:31:53 +0000 |
---|---|---|
committer | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-03-13 16:11:38 +0000 |
commit | 104fbd7b533c40f19465c85e884f10ae500e639e (patch) | |
tree | 10b6fb262fe72e5afeb13ba5b8f491fb3c1a825e /examples/graph_edsr.h | |
parent | 797b76b1aef38ea3be6f68ae2bf323048e9beff8 (diff) | |
download | ComputeLibrary-104fbd7b533c40f19465c85e884f10ae500e639e.tar.gz |
COMPMID-3221: Add DeconvolutionLayerDescriptor
A new struct for DeconvolutionLayerNode is added for better
extendability.
Change-Id: I935277e8073a8295de7b0059b946cb637085f1ff
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2883
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'examples/graph_edsr.h')
-rw-r--r-- | examples/graph_edsr.h | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/examples/graph_edsr.h b/examples/graph_edsr.h index e31cc8940a..cb467d0377 100644 --- a/examples/graph_edsr.h +++ b/examples/graph_edsr.h @@ -1245,13 +1245,14 @@ public: _graph.add_connection(id_pre_upscale_Conv2D_bias, 0, id_pre_upscale_BiasAdd, 2); NodeID id_upscale_net_FakeQuantWithMinMaxVars_1 = _graph.add_node<DeconvolutionLayerNode>( - PadStrideInfo - { - 2, 2, - 0, 0, - 0, 0, - DimensionRoundingType::FLOOR }, - QuantizationInfo{ 0.004990961868315935, 26 }); + descriptors::DeconvolutionLayerDescriptor + { + PadStrideInfo{ + 2, 2, + 0, 0, + 0, 0, + DimensionRoundingType::FLOOR }, + QuantizationInfo{ 0.004990961868315935, 26 } }); INode *node_upscale_net_FakeQuantWithMinMaxVars_1 = _graph.node(id_upscale_net_FakeQuantWithMinMaxVars_1); node_upscale_net_FakeQuantWithMinMaxVars_1->set_common_node_parameters(NodeParams{ "upscale_net_FakeQuantWithMinMaxVars_1", target }); _graph.add_connection(id_pre_upscale_BiasAdd, 0, id_upscale_net_FakeQuantWithMinMaxVars_1, 0); |