fix pad for 3 args

d5939189 · Khalique · 3da18f30 · d5939189
Commit d5939189 authored Aug 26, 2019 by Khalique
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/onnx/onnx.cpp src/onnx/onnx.cpp +1 -1

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -365,7 +365,7 @@ struct onnx_parser
        if(args.size() == 3)
        {
            uint64_t axis = 1;
-            auto l1       = prog.add_instruction(op, args[0], args[1]);
+            auto l1       = prog.add_instruction(op, l0, args[1]);
            auto l2 = prog.add_instruction(op::broadcast{axis, l1->get_shape().lens()}, args[2]);
            return prog.add_instruction(op::add{}, l1, l2);
        }