Hi, thank you for the interesting discussion.
fx records primitive ATen operations and surely it can share op mapping table as torchscript. I don’t think we need to develop & maintian a separate table for fx backend.
@Lyken17, I’m interested in extracting ATen op records from the fx graph. Would you mind sharing your thoughts on how we can do this? I’m new to TorchFx and these information seems to be abstracted away at the FX graph based on my short experience.
jit.trace after fx.symbolic_trace that @masahi suggested could be one way to achieve this, but I’m wondering if there is a way to do this purely within Fx.