set multiple outputs (#4094)

d2c6c27e · baojun · Scott Cyphers · 2e197c84 · d2c6c27e
Commit d2c6c27e authored Dec 20, 2019 by baojun Committed by Scott Cyphers Dec 20, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

matmul.cpp src/ngraph/frontend/fluid/operators/matmul.cpp +1 -0

No files found.
--- a/src/ngraph/frontend/fluid/operators/matmul.cpp
+++ b/src/ngraph/frontend/fluid/operators/matmul.cpp
@@ -263,6 +263,7 @@ void MatMulGrad::pre_validate_and_infer_types()
        get_input_partial_shape(2).is_dynamic())
    {
        set_output_type(0, input_element_type, PartialShape::dynamic());
+        set_output_type(1, input_element_type, PartialShape::dynamic());
    }
 }
 NodeVector MatMulGrad::decompose_op() const