Short name describing what triggered the graph break
autograd.Function.apply: set_output_grad_dtype arity mismatch
Values or code snippet captured at the break point
{len(ctx.output_grad_dtypes)} declarations for {len(fwd_out_vts)} outputs
Explanation of why the graph break was triggered
ctx.set_output_grad_dtype must declare one dtype per forward output.
Hints on how to resolve the graph break