Thanks for responding,
what do you mean by “in-place” in this case ?
lets assume that I have a traced module that I got by: torch.jit.trace(model, sample_inputs).
do you mean that I need to change its “graph” field or “inlined_graph” field to make the module to represent the new graph? (thats mean that all its parameters should be changed if needed)