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 /src/graph/GraphBuilder.cpp | |
| 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 'src/graph/GraphBuilder.cpp')
| -rw-r--r-- | src/graph/GraphBuilder.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/graph/GraphBuilder.cpp b/src/graph/GraphBuilder.cpp index 228f2d211a..89c8c20de8 100644 --- a/src/graph/GraphBuilder.cpp +++ b/src/graph/GraphBuilder.cpp @@ -376,6 +376,11 @@ NodeID GraphBuilder::add_depthwise_convolution_node(Graph &g, NodeParams params, return conv_nid; } +NodeID GraphBuilder::add_dequantization_node(Graph &g, NodeParams params, NodeIdxPair input) +{ + return create_simple_single_input_output_node<DequantizationLayerNode>(g, params, input); +} + NodeID GraphBuilder::add_detection_output_node(Graph &g, NodeParams params, NodeIdxPair input_loc, NodeIdxPair input_conf, NodeIdxPair input_priorbox, const DetectionOutputLayerInfo &detect_info) { check_nodeidx_pair(input_loc, g); |