diff options
Diffstat (limited to 'ModelToINetworkConverter.cpp')
-rw-r--r-- | ModelToINetworkConverter.cpp | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/ModelToINetworkConverter.cpp b/ModelToINetworkConverter.cpp index 24fb4903..0d6ddc34 100644 --- a/ModelToINetworkConverter.cpp +++ b/ModelToINetworkConverter.cpp @@ -59,8 +59,17 @@ void ModelToINetworkConverter<HalPolicy>::Convert() totalPoolSize += pool.size(); } + using NetworkOptions = std::vector<armnn::BackendOptions>; + NetworkOptions networkOptions; + armnn::BackendOptions shapeInferenceMethodOption("ShapeInferenceMethod", + { + { "InferAndValidate", true } + }); + + networkOptions.push_back(shapeInferenceMethodOption); + // Create armnn::INetwork - m_Data.m_Network = armnn::INetwork::Create(); + m_Data.m_Network = armnn::INetwork::Create(networkOptions); // add operations to it // track which layer outputs each operand |