The node you are looking for is at grad_instr_proxy.