Skip to content

T
tic
Commit f5c10e9f by Yizhi Liu Committed by Tianqi Chen
Options

[TF] ignore Truncate in cast (#2022)

parent 26ceaac8
Showing with 2 additions and 1 deletions
nnvm/python/nnvm/frontend/tensorflow.py
...@@ -306,7 +306,8 @@ def _cast(): ...@@ -306,7 +306,8 @@ def _cast():
def _impl(inputs, attr, params): def _impl(inputs, attr, params):
# Convert from tensorflow Dtype to str # Convert from tensorflow Dtype to str
attr['DstT'] = attr['DstT'].name 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 return _impl
def _expand_dims(): def _expand_dims():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment