[ONNX] Addition of diagnostic tool API (#43020)
Summary:
Added initial diagnostic tool API
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43020
Reviewed By: malfet
Differential Revision: D23398459
Pulled By: bzinodev
fbshipit-source-id: 7a6d9164a19e3ba51676fbcf645c4d358825eb42