I suppose that the optimization process that you have mentioned is the transform passes uring the BuildRelay() process, if so, you can print the IR before and after each transform pass and compare the difference
I suppose that the optimization process that you have mentioned is the transform passes uring the BuildRelay() process, if so, you can print the IR before and after each transform pass and compare the difference