Hi, I am trying to get the intermediate buffers between layers in a graph module. I am trying to accomplish something similiar to TVM’s get_calibration_data (tvm.relay.analysis — tvm 0.15.dev0 documentation), however for a specific function, and not between sub-graphs.
Any ideas how I can do that?
Thanks!