diff --git a/nnvm/python/nnvm/frontend/tensorflow.py b/nnvm/python/nnvm/frontend/tensorflow.py index 28877ef19e0e..da6fcc94bf6f 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():