diff options
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/INetwork.hpp | 4 | ||||
| -rw-r--r-- | include/armnn/TypesUtils.hpp | 24 |
2 files changed, 22 insertions, 6 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index 349c7e87b5..ecc888d9c1 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -801,8 +801,10 @@ public: size_t GetNumInputs() const; size_t GetNumOutputs() const; + void ExecuteStrategy(IStrategy& strategy) const; + // Creates a copy of the IOptimizedNetwork. The IOptimizedNetwork will not be reoptimized, - // the provided ModelOptions will only be used when creating a LoadedNetwork. + // the provided ModelOptions will only be used when creating a LoadedNetwork. IOptimizedNetwork(const IOptimizedNetwork& other, const ModelOptions& modelOptions); IOptimizedNetwork(std::unique_ptr<Graph> graph); IOptimizedNetwork(std::unique_ptr<OptimizedNetworkImpl> impl); diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index ccb0280457..ca7e7c58ac 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -301,16 +301,30 @@ inline std::ostream& operator<<(std::ostream& os, Status stat) } -inline std::ostream & operator<<(std::ostream & os, const armnn::TensorShape & shape) +inline std::ostream& operator<<(std::ostream& os, const armnn::TensorShape& shape) { os << "["; - for (uint32_t i=0; i<shape.GetNumDimensions(); ++i) + if (shape.GetDimensionality() != Dimensionality::NotSpecified) { - if (i!=0) + for (uint32_t i = 0; i < shape.GetNumDimensions(); ++i) { - os << ","; + if (i != 0) + { + os << ","; + } + if (shape.GetDimensionSpecificity(i)) + { + os << shape[i]; + } + else + { + os << "?"; + } } - os << shape[i]; + } + else + { + os << "Dimensionality Not Specified"; } os << "]"; return os; |