Hi, you need 2 Ops, first for forward, 2d for backward.
Good example is
Eigh forward https://github.com/Theano/Theano/blob/master/theano/tensor/nlinalg.py#L293
Eigh backward https://github.com/Theano/Theano/blob/master/theano/tensor/nlinalg.py#L365