diff options
Diffstat (limited to 'src/armnnTfParser/TfParser.hpp')
| -rw-r--r-- | src/armnnTfParser/TfParser.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.hpp b/src/armnnTfParser/TfParser.hpp index 0d1e497e29..20c5233c8e 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -10,6 +10,7 @@ #include "armnn/Tensor.hpp" #include "armnn/INetwork.hpp" +#include <list> #include <map> #include <memory> #include <unordered_map> @@ -241,6 +242,8 @@ private: /// Map of TensorFlow operation names to parsing member functions. static const std::map<std::string, OperationParsingFunction> ms_OperationNameToParsingFunctions; + static const std::list<std::string> m_ControlInputs; + std::map<std::string, armnn::TensorShape> m_InputShapes; std::vector<std::string> m_RequestedOutputs; |