diff options
author | Finn Williams <Finn.Williams@arm.com> | 2021-02-11 16:16:42 +0000 |
---|---|---|
committer | KeithARM <keith.davis@arm.com> | 2021-02-15 15:04:31 +0000 |
commit | 406463269f55a5baefb941b51e10f423f6d3250a (patch) | |
tree | 2299b75c47cae44d9c8be4bf670f30c9d65a01a8 /delegate/src | |
parent | b55d366a93d8d3699ce5e906134c695c8040c27e (diff) | |
download | armnn-406463269f55a5baefb941b51e10f423f6d3250a.tar.gz |
IVGCVSW-5686 Add GpuAcc MLGO tuning file configuration argument
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I3f320499c379162f9d1b00cc8816bd144cd7eee4
Diffstat (limited to 'delegate/src')
-rw-r--r-- | delegate/src/armnn_external_delegate.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/delegate/src/armnn_external_delegate.cpp b/delegate/src/armnn_external_delegate.cpp index 4dba07d3b8..edf46efb98 100644 --- a/delegate/src/armnn_external_delegate.cpp +++ b/delegate/src/armnn_external_delegate.cpp @@ -125,6 +125,11 @@ TfLiteDelegate* tflite_plugin_create_delegate(char** options_keys, armnn::BackendOptions option("GpuAcc", {{"TuningLevel", atoi(options_values[i])}}); options.AddBackendOption(option); } + else if (std::string(options_keys[i]) == std::string("gpu-mlgo-tuning-file")) + { + armnn::BackendOptions option("GpuAcc", {{"MLGOTuningFilePath", std::string(options_values[i])}}); + options.AddBackendOption(option); + } else if (std::string(options_keys[i]) == std::string("gpu-tuning-file")) { armnn::BackendOptions option("GpuAcc", {{"TuningFile", std::string(options_values[i])}}); |