Fuse two mul scalar in prelu

JDAI-CV · Jan 10, 2019 · 8c515a6 · 8c515a6
1 parent 721c078
commit 8c515a6
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/tools/onnx2daq/OnnxConverter.cpp b/tools/onnx2daq/OnnxConverter.cpp
@@ -269,18 +269,15 @@ void OnnxConverter::Convert(const ONNX_NAMESPACE::ModelProto &model_proto, const
             const auto imm2_name = output_name + "_imm2";
             const auto imm3_name = output_name + "_imm3";
             const auto imm4_name = output_name + "_imm4";
-            const auto imm5_name = output_name + "_imm5";
-            const auto imm6_name = output_name + "_imm6";
             if (onnx_tensors_[slope_name].shape != Shape{1}) {
                 // TODO: support it
                 throw std::invalid_argument("Only support one element slope.");
             }
             addLayerRelu(input_name, imm1_name);
-            addLayerMul(input_name, onnx_tensors_[slope_name].data[0], imm2_name);
-            addLayerMul(imm2_name, -1.f, imm3_name);
-            addLayerRelu(imm3_name, imm4_name);
-            addLayerMul(imm4_name, -1.f, imm5_name);
-            addLayerAdd(imm1_name, imm5_name, output_name);
+            addLayerMul(input_name, -onnx_tensors_[slope_name].data[0], imm2_name);
+            addLayerRelu(imm2_name, imm3_name);
+            addLayerMul(imm3_name, -1.f, imm4_name);
+            addLayerAdd(imm1_name, imm4_name, output_name);
             // TODO:
             LOG(INFO) << "Converting PRelu completed";
         } else if (op == "Add") {