fix

pfnet-research · Apr 17, 2019 · 4d22529 · 4d22529
1 parent fec280c
commit 4d22529
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/menoh/attribute_completion_and_shape_inference.hpp b/menoh/attribute_completion_and_shape_inference.hpp
@@ -647,7 +647,7 @@ if(transB) {
 
 if(a_dims.at(1) != b_dims.at(0)) {
     throw dimension_mismatch(
-        node.op_type, input(0), "trans(A)[1] and trans(B)[0])",
+        node.op_type, output(0), "trans(A)[1] and trans(B)[0])",
         std::to_string(a_dims.at(1)), std::to_string(b_dims.at(0)));
 }
 

diff --git a/scripts/gen_attribute_completion_and_shape_inference_hpp.py b/scripts/gen_attribute_completion_and_shape_inference_hpp.py
@@ -299,7 +299,7 @@ def main():
 
 if(a_dims.at(1) != b_dims.at(0)) {
     throw dimension_mismatch(
-        node.op_type, input(0), "trans(A)[1] and trans(B)[0])",
+        node.op_type, output(0), "trans(A)[1] and trans(B)[0])",
         std::to_string(a_dims.at(1)), std::to_string(b_dims.at(0)));
 }