Joe

icon indicating an email [email protected]

icon location Tokyo, Japan icon location LLVM @ Meta AI. Previously Compiler @ PFN.

Results 13 comments of Joe

Generate name for each member of list arg

My idea: always unroll args to follow `torch.onnx.export` style. For PFTO, we wrap the module for list inputs (future work).

Why is there no backward for spspmm in the latest version?

I'm looking forward to seeing this feature soon :)

Generate name for each member of list arg

memo: Also support list, tuple outputs