/*
   Add custom ops to the end.
*/
DEF_OP(CUSTOM_SOFTMAX)