Fix the elementwise_ops axis bug, make elementwise_ops has the same the op function for override ops