Oneflow-Inc / one-fx

A toolkit for developers to simplify the transformation of nn.Module instances. It's now corresponding to Pytorch.fx.
Other
13 stars 1 forks source link

Add set_arg_for_node for Interpreter. #20

Closed AsakusaRinne closed 1 year ago

AsakusaRinne commented 1 year ago

Add set_arg_for_node for Interpreter to help to support auto placement. https://github.com/Oneflow-Inc/OneTeam/issues/1917