diff options
Diffstat (limited to 'src/armnn/Graph.hpp')
| -rw-r--r-- | src/armnn/Graph.hpp | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/src/armnn/Graph.hpp b/src/armnn/Graph.hpp index 87e0da826f..731ae1e5aa 100644 --- a/src/armnn/Graph.hpp +++ b/src/armnn/Graph.hpp @@ -13,6 +13,7 @@ #include <armnn/Exceptions.hpp> #include <armnn/utility/Assert.hpp> #include <armnn/utility/PolymorphicDowncast.hpp> +#include <armnn/utility/TransformIterator.hpp> #include <list> #include <map> @@ -20,8 +21,6 @@ #include <unordered_set> #include <vector> -#include <boost/iterator/transform_iterator.hpp> - namespace armnn { @@ -51,9 +50,9 @@ public: using Iterator = LayerList::const_iterator; // Const so pointers in the list can't be modified externally. using IteratorDifference = Iterator::difference_type; - using ConstIterator = boost::transform_iterator<decltype(&PtrCast<const Layer>), Iterator>; - using ConstIteratorInputs = boost::transform_iterator<decltype(&PtrCast<const InputLayer>), Iterator>; - using ConstIteratorOutputs = boost::transform_iterator<decltype(&PtrCast<const OutputLayer>), Iterator>; + using ConstIterator = TransformIterator<decltype(&PtrCast<const Layer>), Iterator>; + using ConstIteratorInputs = TransformIterator<decltype(&PtrCast<const InputLayer>), Iterator>; + using ConstIteratorOutputs = TransformIterator<decltype(&PtrCast<const OutputLayer>), Iterator>; /// Wrapper class returned by Graph::GetInputLayers() struct InputLayersAccessor |