[incubator-tvm] branch master updated: [RELAY][FRONTEND][TF] Fix FuseBatchNorm output cast error if need_cast is True (#4894)

[kevinthesun]

This is an automated email from the ASF dual-hosted git repository.

kevinthesun pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git


The following commit(s) were added to refs/heads/master by this push:
     new fccf226  [RELAY][FRONTEND][TF] Fix FuseBatchNorm output cast error if 
need_cast is True (#4894)
fccf226 is described below

commit fccf2268c857127739fadb0349ef4bafaf66c282
Author: hcyang <yh...@126.com>
AuthorDate: Wed Feb 19 14:33:16 2020 +0800

    [RELAY][FRONTEND][TF] Fix FuseBatchNorm output cast error if need_cast is 
True (#4894)
---
 python/tvm/relay/frontend/tensorflow.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/tvm/relay/frontend/tensorflow.py 
b/python/tvm/relay/frontend/tensorflow.py
index f920682..587b076 100644
--- a/python/tvm/relay/frontend/tensorflow.py
+++ b/python/tvm/relay/frontend/tensorflow.py
@@ -897,6 +897,7 @@ def _fused_batch_norm():
                       disables=['momentum'])(inputs, attr)
 
         if need_cast:
+            out = _expr.TupleGetItem(out.astuple(), 0)
             out = _op.cast(out, dtype=attr['T'].name)
         return out
     return _impl