Add more requirements for the three operators to ensure it is correct when input is transposed.

29865b9a · Shucai Xiao · 7c3b9d48 · 29865b9a · 29865b9a · 29865b9a
Commit 29865b9a authored Apr 17, 2019 by Shucai Xiao
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

src/targets/gpu/lrn.cpp src/targets/gpu/lrn.cpp +1 -1

src/targets/gpu/relu.cpp src/targets/gpu/relu.cpp +1 -1

src/targets/gpu/tanh.cpp src/targets/gpu/tanh.cpp +1 -1

No files found.
--- a/src/targets/gpu/lrn.cpp
+++ b/src/targets/gpu/lrn.cpp
@@ -7,7 +7,7 @@ namespace gpu {

 shape miopen_lrn::compute_shape(const std::vector<shape>& inputs) const
 {
-    check_shapes{inputs, *this}.has(2).not_broadcasted();
+    check_shapes{inputs, *this}.has(2).standard();
    return inputs.at(1);
 }


--- a/src/targets/gpu/relu.cpp
+++ b/src/targets/gpu/relu.cpp
@@ -7,7 +7,7 @@ namespace gpu {

 shape miopen_relu::compute_shape(const std::vector<shape>& inputs) const
 {
-    check_shapes{inputs, *this}.has(2).not_broadcasted();
+    check_shapes{inputs, *this}.has(2).not_broadcasted().not_transposed();
    return inputs.at(1);
 }


--- a/src/targets/gpu/tanh.cpp
+++ b/src/targets/gpu/tanh.cpp
@@ -7,7 +7,7 @@ namespace gpu {

 shape miopen_tanh::compute_shape(const std::vector<shape>& inputs) const
 {
-    check_shapes{inputs, *this}.has(2).not_broadcasted();
+    check_shapes{inputs, *this}.has(2).not_broadcasted().not_transposed();
    return inputs.at(1);
 }