diff options
author | 박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com> | 2019-09-05 17:02:19 +0900 |
---|---|---|
committer | GitHub Enterprise <noreply-CODE@samsung.com> | 2019-09-05 17:02:19 +0900 |
commit | f2b2cdf0965073f6e5eb295debd98470ff9caaeb (patch) | |
tree | 1414ad74a8306246b92c70b030da83363c836142 /compiler | |
parent | 8e990bfb347bdbf5f80f2344ecdde82b36fff69a (diff) | |
download | nnfw-f2b2cdf0965073f6e5eb295debd98470ff9caaeb.tar.gz nnfw-f2b2cdf0965073f6e5eb295debd98470ff9caaeb.tar.bz2 nnfw-f2b2cdf0965073f6e5eb295debd98470ff9caaeb.zip |
[moco-tf] Skip FuseBinaryIntoPreceding on Scalar (#7219)
The current implementation of FuseBinaryIntoPreceding gets stuck when
it encounters scalar constants.
Let's make FuseBinaryIntoPreceding skip on this case.
Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
Diffstat (limited to 'compiler')
-rw-r--r-- | compiler/moco-tf/src/Transforms/FuseBinaryIntoPreceding.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/compiler/moco-tf/src/Transforms/FuseBinaryIntoPreceding.cpp b/compiler/moco-tf/src/Transforms/FuseBinaryIntoPreceding.cpp index 083bc2f63..bfaf910b4 100644 --- a/compiler/moco-tf/src/Transforms/FuseBinaryIntoPreceding.cpp +++ b/compiler/moco-tf/src/Transforms/FuseBinaryIntoPreceding.cpp @@ -453,6 +453,11 @@ bool fuse_to_preceding(loco::Graph *graph, moco::tf::TFAdd *node) } // TODO do something when rank() is 0 + if (addparam_shape.rank() == 0) + { + // Not supported yet + return false; + } assert(addparam_shape.rank() != 0); // TODO support FullyConnected |