Function _e_square_grad

operators.py:219–221

(op, grad)

    return trainer_ops.e_square(inputs,eid_low=(eid&0xFFFFFFFF),eid_high=(eid>>32),times=times)
@ops.RegisterGradient("ESquare")
def _e_square_grad(op, grad):
    with tf.name_scope("ESquareGrad"), tf.xla.experimental.jit_scope(compile_ops=False):
        return trainer_ops.e_square_grad(grad,op.inputs[0],eid_low=op.get_attr("eid_low"), eid_high=op.get_attr("eid_high"),times = op.get_attr("times"))

def e_reduce_max_one(inputs):
    global eid,times