tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [incubator-tvm] t-vi edited a comment on pull request #5756: make dtype="float" mean float32 for relay.expr.const
Date Wed, 10 Jun 2020 17:00:28 GMT

t-vi edited a comment on pull request #5756:
URL: https://github.com/apache/incubator-tvm/pull/5756#issuecomment-642134657


   @tqchen The thing I'm finding awkward to use is
   ```python
   >>> tvm.relay.expr.const(1, dtype="float").data.dtype                        
                                                                                         
                                       
   'float64'
   
   >>> tvm.relay.expr.const(1.).data.dtype                                      
                                                                                         
                                       
   'float32'
   
   >>> tvm.relay.expr.var("x", "float")                                         
                                                                                         
                                       
   Var(x, ty=TensorType([], float32))
   ```
   I noticed because the `relay.frontend.from_pytorch` uses float to mean float32 in many
places.
   Alternatively, we might fix `var` but at least the PyTorch frontend would need fixing.
   


----------------------------------------------------------------
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



Mime
View raw message