diff options
Diffstat (limited to 'src/graph/GraphBuilder.cpp')
-rw-r--r-- | src/graph/GraphBuilder.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/graph/GraphBuilder.cpp b/src/graph/GraphBuilder.cpp index 72c45fae30..01520421f0 100644 --- a/src/graph/GraphBuilder.cpp +++ b/src/graph/GraphBuilder.cpp @@ -376,6 +376,12 @@ NodeID GraphBuilder::add_depthwise_convolution_node(Graph &g, NodeParams params, return conv_nid; } + +NodeID GraphBuilder::add_depth_to_space_node(Graph &g, NodeParams params, NodeIdxPair input, int32_t block_shape) +{ + return create_simple_single_input_output_node<DepthToSpaceLayerNode>(g, params, input, block_shape); +} + NodeID GraphBuilder::add_dequantization_node(Graph &g, NodeParams params, NodeIdxPair input) { return create_simple_single_input_output_node<DequantizationLayerNode>(g, params, input); |