Set enable-sep-transposed-conv for model compile

compile model with enable-sep-transposed-conv=true Signed-off-by: Keyon Jie <[email protected]>
intel · Feb 9, 2024 · e67b877 · e67b877
1 parent d8c1022
commit e67b877
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/IENetwork.cpp b/IENetwork.cpp
@@ -38,11 +38,10 @@ bool IENetwork::loadNetwork() {
 
     ALOGD("Creating infer request for Intel Device Type : %s", deviceStr.c_str());
     if (mNetwork) {
-//        compiled_model = ie.compile_model(mNetwork, deviceStr);
         ALOGD("LoadNetwork is done....");
 
 	ov::AnyMap config;
-	config["NPU_COMPILATION_MODE_PARAMS"] = "enable-se-ptrs-operations=true";
+	config["NPU_COMPILATION_MODE_PARAMS"] = "enable-se-ptrs-operations=true enable-sep-transposed-conv=true";
 
 	compiled_model = ie.compile_model(mNetwork, deviceStr, config);