diff options
Diffstat (limited to 'arm_compute/graph/frontend/IStreamOperators.h')
| -rw-r--r-- | arm_compute/graph/frontend/IStreamOperators.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/arm_compute/graph/frontend/IStreamOperators.h b/arm_compute/graph/frontend/IStreamOperators.h index 1eb6522935..350d78fd1c 100644 --- a/arm_compute/graph/frontend/IStreamOperators.h +++ b/arm_compute/graph/frontend/IStreamOperators.h @@ -48,6 +48,18 @@ inline IStream &operator<<(IStream &s, ILayer &&layer) s.add_layer(layer); return s; } +/** Overloaded stream operator to add a node to the graph + * + * @param[in, out] s Stream to add the tensor + * @param[in] layer Layer to be added + * + * @return Updated stream + */ +inline IStream &operator<<(IStream &s, ILayer &layer) +{ + s.add_layer(layer); + return s; +} /** Overloaded stream operator to provide a target hint to the graph * * @param[in, out] s Stream to provide the hint to |