diff options
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 |