diff options
author | Isabella Gottardi <isabella.gottardi@arm.com> | 2019-11-05 17:50:27 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-11-14 13:32:30 +0000 |
commit | cd4e9abf7a165f15ccd10ac4541365d4f8a6db19 (patch) | |
tree | 442cb14065b866065c1e2837490ae36731a38eef /arm_compute/graph/GraphBuilder.h | |
parent | ee8cf3287265436d0d12effa36e876dc251eee8c (diff) | |
download | ComputeLibrary-cd4e9abf7a165f15ccd10ac4541365d4f8a6db19.tar.gz |
COMPMID-2452: Add mnist example
* Add small-mnist example
* Add PrintAccessor
* Add DequantizationLayer graph node
Change-Id: I7bc8011e5a602f40fa3c47b231a2a69c804e78c2
Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2274
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'arm_compute/graph/GraphBuilder.h')
-rw-r--r-- | arm_compute/graph/GraphBuilder.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/arm_compute/graph/GraphBuilder.h b/arm_compute/graph/GraphBuilder.h index dc41ed5367..c1c56c3dd2 100644 --- a/arm_compute/graph/GraphBuilder.h +++ b/arm_compute/graph/GraphBuilder.h @@ -205,6 +205,15 @@ public: * @return Node ID of the created node, EmptyNodeID in case of error */ static NodeID add_elementwise_node(Graph &g, NodeParams params, NodeIdxPair input0, NodeIdxPair input1, EltwiseOperation operation); + /** Adds a dequantization node to the graph + * + * @param[in] g Graph to add the node to + * @param[in] params Common node parameters + * @param[in] input Input to the dequantization node as a NodeID-Index pair + * + * @return Node ID of the created node, EmptyNodeID in case of error + */ + static NodeID add_dequantization_node(Graph &g, NodeParams params, NodeIdxPair input); /** Adds a detection output layer node to the graph * * @param[in] g Graph to add the node to |