The node you are looking for is at grad_instr_call.