From c9066bafac567d7d003d433125472a300d0ac5b9 Mon Sep 17 00:00:00 2001 From: Yizhi Liu Date: Sun, 28 Oct 2018 01:21:59 -0700 Subject: [PATCH] [TF] ignore Truncate in cast --- nnvm/python/nnvm/frontend/tensorflow.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nnvm/python/nnvm/frontend/tensorflow.py b/nnvm/python/nnvm/frontend/tensorflow.py index ad7c4fc6796f..9cd07cca3cc6 100644 --- a/nnvm/python/nnvm/frontend/tensorflow.py +++ b/nnvm/python/nnvm/frontend/tensorflow.py @@ -306,7 +306,8 @@ def _cast(): def _impl(inputs, attr, params): # Convert from tensorflow Dtype to str attr['DstT'] = attr['DstT'].name - return AttrCvt(op_name='cast', transforms={'DstT': 'dtype'}, ignores=['SrcT'])(inputs, attr) + return AttrCvt(op_name='cast', transforms={'DstT': 'dtype'}, + ignores=['SrcT', 'Truncate'])(inputs, attr) return _impl def _expand_dims():