diff options
Diffstat (limited to 'delegate/src')
-rw-r--r-- | delegate/src/DelegateOptions.cpp | 18 | ||||
-rw-r--r-- | delegate/src/armnn_delegate.cpp | 2 |
2 files changed, 19 insertions, 1 deletions
diff --git a/delegate/src/DelegateOptions.cpp b/delegate/src/DelegateOptions.cpp index 9413a4689a..f3e13c90c6 100644 --- a/delegate/src/DelegateOptions.cpp +++ b/delegate/src/DelegateOptions.cpp @@ -156,6 +156,24 @@ DelegateOptions::DelegateOptions(char const* const* options_keys, { optimizerOptions.m_Debug = armnn::stringUtils::StringToBool(options_values[i]); } + // Infer output-shape + else if (std::string(options_keys[i]) == std::string("infer-output-shape")) + { + armnn::BackendOptions backendOption("ShapeInferenceMethod", + { + { "InferAndValidate", armnn::stringUtils::StringToBool(options_values[i]) } + }); + optimizerOptions.m_ModelOptions.push_back(backendOption); + } + // Allow expanded dims + else if (std::string(options_keys[i]) == std::string("allow-expanded-dims")) + { + armnn::BackendOptions backendOption("AllowExpandedDims", + { + { "AllowExpandedDims", armnn::stringUtils::StringToBool(options_values[i]) } + }); + optimizerOptions.m_ModelOptions.push_back(backendOption); + } // Process memory-import else if (std::string(options_keys[i]) == std::string("memory-import")) { diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp index 4d71f26b09..6e1a91f9e4 100644 --- a/delegate/src/armnn_delegate.cpp +++ b/delegate/src/armnn_delegate.cpp @@ -308,7 +308,7 @@ ArmnnSubgraph* ArmnnSubgraph::Create(TfLiteContext* tfLiteContext, DelegateData delegateData(delegate->m_Options.GetBackends()); // Build ArmNN Network - armnn::NetworkOptions networkOptions = {}; + armnn::NetworkOptions networkOptions = delegate->m_Options.GetOptimizerOptions().m_ModelOptions; armnn::NetworkId networkId; delegateData.m_Network = armnn::INetwork::Create(networkOptions); |