diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-05-01 15:26:20 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:51:17 +0000 |
commit | 5c2fb3f34462632b99331e2cc2d964c99fc1782b (patch) | |
tree | 16ee3edc412fcf7e3d20241ca8fb093d9774863d /arm_compute/graph/frontend/IStreamOperators.h | |
parent | cac13b1cfd593889271f8e2191be2039b8d88f36 (diff) | |
download | ComputeLibrary-5c2fb3f34462632b99331e2cc2d964c99fc1782b.tar.gz |
COMPMID-997: Add support for node's name in GraphAPI.
Change-Id: I0ca02e42807c1ad9afeffb7202a3556feb11442f
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/129701
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
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 |