Merge branch 'develop' of https://github.com/ROCmSoftwarePlatform/AMDMIGraphX into tf_pb_py

21991697 · Khalique · 7faa706b · 154f21f6 · 21991697 · 21991697
Commit 21991697 authored Mar 11, 2019 by Khalique
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

src/tf/tf.cpp src/tf/tf.cpp +1 -1

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -157,7 +157,7 @@ struct onnx_parser
    template <class T>
    instruction_ref add_broadcastable_binary_op(instruction_ref arg0, instruction_ref arg1, T x)
    {
-        if(arg0->get_shape() != arg1->get_shape())
+        if(arg0->get_shape().lens() != arg1->get_shape().lens())
        {
            // Example:
            // s0 = (3,2,4,5) and s1 = (2,1,1)

--- a/src/tf/tf.cpp
+++ b/src/tf/tf.cpp
@@ -167,7 +167,7 @@ struct tf_parser
    template <class T>
    instruction_ref add_broadcastable_binary_op(instruction_ref arg0, instruction_ref arg1, T x)
    {
-        if(arg0->get_shape() != arg1->get_shape())
+        if(arg0->get_shape().lens() != arg1->get_shape().lens())
        {
            // Example:
            // s0 = (3,2,4,5) and s1 = (2,1,1)