vollo_torch.fx.nnir
Translate Torch models to Vollo compiler NNIR models
- vollo_torch.fx.nnir.to_nnir(module: torch.fx.graph_module.GraphModule)
Note: Expects
module
to have gone throughtorch.fx.passes.shape_prop.ShapeProp()
.Does not support
inplace
or_out
variants of torch modules and functions.