Merge branch 'adjust_gpu_allocation' into ins_fp32_fp16

9a43b32c · Shucai Xiao · 5856e5d0 · 29865b9a · 9a43b32c · 9a43b32c
Commit 9a43b32c authored Apr 17, 2019 by Shucai Xiao
Hide 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);
 }