@gof.local_optimizer([csm_grad(None)])
def local_csm_grad_c(node):
 if (node.op == csm_grad(None)):
  return [csm_grad_c(*node.inputs)]
 return False

@ gof . local_optimizer ( [ csm_grad ( None ) ] ) def local_csm_grad_c ( node ) : if ( node . op == csm_grad ( None ) ) : return [ csm_grad_c ( * node . inputs ) ] return False
csm_grad -> csm_grad_c .