diff options
Diffstat (limited to 'src/armnnDeserializeParser/DeserializeParser.hpp')
-rw-r--r-- | src/armnnDeserializeParser/DeserializeParser.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnnDeserializeParser/DeserializeParser.hpp b/src/armnnDeserializeParser/DeserializeParser.hpp index 9edd959220..ce343dc528 100644 --- a/src/armnnDeserializeParser/DeserializeParser.hpp +++ b/src/armnnDeserializeParser/DeserializeParser.hpp @@ -42,7 +42,7 @@ public: public: // testable helpers - static GraphPtr LoadGraphFromFile(const char* fileName); + static GraphPtr LoadGraphFromFile(const char* fileName, std::string& fileContent); static GraphPtr LoadGraphFromBinary(const uint8_t* binaryContent, size_t len); static TensorRawPtrVector GetInputs(const GraphPtr& graph, unsigned int layerIndex); static TensorRawPtrVector GetOutputs(const GraphPtr& graph, unsigned int layerIndex); @@ -82,6 +82,10 @@ private: GraphPtr m_Graph; std::vector<LayerParsingFunction> m_ParserFunctions; + /// This holds the data of the file that was read in from CreateNetworkFromBinaryFile + /// Needed for m_Graph to point to + std::string m_FileContent; + /// A mapping of an output slot to each of the input slots it should be connected to /// The outputSlot is from the layer that creates this tensor as one of its outputs /// The inputSlots are from the layers that use this tensor as one of their inputs |