diff options
author | Przemyslaw Szczepaniak <pszczepaniak@google.com> | 2021-06-29 12:47:22 +0000 |
---|---|---|
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2021-06-29 12:47:22 +0000 |
commit | 0a9138e1a17e4b0840c72d393c3c2e55071fa5f6 (patch) | |
tree | e0cce7aa384151fefb456aba81f90ab48a4f0f47 | |
parent | 06a6a406bf20de065ba60d27879666e6cf19ab2d (diff) | |
parent | eaf4dc113c9c4af439db451dfef482c9b5e58a72 (diff) | |
download | NeuralNetworks-0a9138e1a17e4b0840c72d393c3c2e55071fa5f6.tar.gz |
Merge "Set shim converted model relaxComputationF32ToF16." into sc-dev am: eaf4dc113c
Original change: https://googleplex-android-review.googlesource.com/c/platform/packages/modules/NeuralNetworks/+/15129331
Change-Id: I40fb6694a9a66b53bbc44db7280b07f706307ad8
-rw-r--r-- | shim_and_sl/ShimConverter.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/shim_and_sl/ShimConverter.cpp b/shim_and_sl/ShimConverter.cpp index 61eb40593..398cc9b9a 100644 --- a/shim_and_sl/ShimConverter.cpp +++ b/shim_and_sl/ShimConverter.cpp @@ -58,6 +58,8 @@ ANeuralNetworksModel* convertSubgraphFromHAL( const auto& subgraph = subgraphIndex == 0 ? model.main : model.referenced[subgraphIndex - 1]; ::android::nn::sl_wrapper::Model resultModel(nnapi); + resultModel.relaxComputationFloat32toFloat16(model.relaxComputationFloat32toFloat16); + auto getExtensionName = [&](uint16_t prefix) -> const std::string* { for (const auto& nameToPrefix : model.extensionNameToPrefix) { if (prefix == nameToPrefix.prefix) { |