diff options
Diffstat (limited to 'include/xnnpack.h')
-rw-r--r-- | include/xnnpack.h | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/include/xnnpack.h b/include/xnnpack.h index 96a438622..646bda905 100644 --- a/include/xnnpack.h +++ b/include/xnnpack.h @@ -862,6 +862,21 @@ enum xnn_status xnn_define_hardswish( uint32_t output_id, uint32_t flags); +/// Define a Leaky ReLU Node and add it to a Subgraph. +/// +/// @param subgraph - a Subgraph object that will own the created Node. +/// @param negative_slope - scale factor for negative input elements. +/// @param input_id - Value ID for the input tensor. The input tensor must be defined in the @a subgraph. +/// @param output_id - Value ID for the output tensor. The output tensor must be defined in the @a subgraph, and its +/// shape must match the shape of the input tensor. +/// @param flags - binary features of the Leaky ReLU Node. No supported flags are currently defined. +enum xnn_status xnn_define_leaky_relu( + xnn_subgraph_t subgraph, + float negative_slope, + uint32_t input_id, + uint32_t output_id, + uint32_t flags); + /// Define a Negate Node and add it to a Subgraph. /// /// @param subgraph - a Subgraph object that will own the created Node. |