Merge pull request #206 from ROCmSoftwarePlatform/multibcast_check

remove unnecessary multibroadcasts

Merge pull request #206 from ROCmSoftwarePlatform/multibcast_check
remove unnecessary multibroadcasts
154f21f6 · Paul Fultz II · GitHub · 3499ec7d · 16fc0314 · 154f21f6
Unverified Commit 154f21f6 authored Mar 09, 2019 by Paul Fultz II Committed by GitHub Mar 09, 2019
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)