tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [incubator-tvm] heliqi commented on issue #4262: [RELAY][Bug] 'name_hint' AttributeError issue when covert tensorflow to TVM
Date Thu, 12 Mar 2020 02:33:18 GMT
heliqi commented on issue #4262: [RELAY][Bug] 'name_hint' AttributeError issue when covert
tensorflow to TVM
URL: https://github.com/apache/incubator-tvm/issues/4262#issuecomment-597975927
 
 
   **Tensorflow error code:**
   ```
   def _get_param(params, input_node):
       if isinstance(input_node, _expr.Constant):
           return np.atleast_1d(input_node.data.asnumpy())
       # input_node is Call,no name_hint
       return params.pop(input_node.name_hint).asnumpy()
   
   def _get_num_param(params, input_node):
       return _get_param(params, input_node).item()
   
   def _get_list_param(params, input_node):
       return _get_param(params, input_node).tolist()
   ```
   
   ```
   def _tile():
       def _impl(inputs, attr, params):
           #inputs.pop() -> CallNode( Op(expand_dim),[CallNode(Op(not_equal),[...])],relay.attrs.ExpandDimsAttrs(0x184d9408),[])
           reps = _get_list_param(params, inputs.pop())
           ......
   ```
   
   I might need the op that prints the tensorflow model to know what param is needed here。

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services

Mime
View raw message