vollo_torch.fx.nnir

Translate Torch models to Vollo compiler NNIR models

vollo_torch.fx.nnir.to_nnir(module: GraphModule)

Note: Expects module to have gone through torch.fx.passes.shape_prop.ShapeProp().

Does not support inplace or _out variants of torch modules and functions.