Add prim name to error message for operators in nn_ops.py