diff options
Diffstat (limited to 'reference_model/src/ops')
-rw-r--r-- | reference_model/src/ops/ewise_unary.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/reference_model/src/ops/ewise_unary.cc b/reference_model/src/ops/ewise_unary.cc index 041bbdb..13e517b 100644 --- a/reference_model/src/ops/ewise_unary.cc +++ b/reference_model/src/ops/ewise_unary.cc @@ -48,9 +48,9 @@ int UnaryNode<Rank, Dtype>::checkTensorAttributes() } // output and input must be the same types - if (inputs[0]->matchRankSize(*outputs[0])) + if (inputs[0]->matchRankTypeShape(*outputs[0])) { - printNodeValidationError("UnaryNode: input and output rank must match"); + printNodeValidationError("UnaryNode: input and output rank/type/shape must match"); return 1; } |