Skip to content
This repository has been archived by the owner on Jul 18, 2024. It is now read-only.

Commit

Permalink
Compile model with params suggested from the NPU team
Browse files Browse the repository at this point in the history
compile model with enable-sep-transposed-conv=true and
enable-se-ptrs-operations=true

Signed-off-by: Keyon Jie <[email protected]>
  • Loading branch information
keyonjie authored and anikulk committed Feb 14, 2024
1 parent 0328bfd commit 6c5aa71
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion IENetwork.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,9 @@ bool IENetwork::loadNetwork() {

ALOGD("Creating infer request for Intel Device Type : %s", deviceStr.c_str());
if (mNetwork) {
compiled_model = ie.compile_model(mNetwork, deviceStr);
ov::AnyMap config;
config["NPU_COMPILATION_MODE_PARAMS"] = "enable-se-ptrs-operations=true enable-sep-transposed-conv=true";
compiled_model = ie.compile_model(mNetwork, deviceStr, config);
ALOGD("LoadNetwork is done....");

#if __ANDROID__
Expand Down

0 comments on commit 6c5aa71

Please sign in to comment.