Solved by albanD in post #2 
              
              
                Hi, 
You will need to specify it in tools/autograd/derivatives.yaml then recompile. 
You can check the entry for relu there and the comment at the beginning of the file for more details.