Function _e_softmax_cross_entropy_grad

operators.py:111–113

(op,grad)

    return trainer_ops.e_softmax_cross_entropy(inputs,real,eid_low=(eid&0xFFFFFFFF),eid_high=(eid>>32),times=times)
@ops.RegisterGradient("ESoftmaxCrossEntropy")
def _e_softmax_cross_entropy_grad(op,grad):
    with tf.name_scope("ESoftmaxCrossEntropyGrad"), tf.xla.experimental.jit_scope(compile_ops=False):
        return trainer_ops.e_softmax_cross_entropy_grad(grad,op.inputs[0],op.inputs[1],eid_low=op.get_attr("eid_low"), eid_high=op.get_attr("eid_high"),times = op.get_attr("times"))[0],trainer_ops.e_softmax_cross_entropy_grad(grad,op.inputs[0],op.inputs[1],eid_low=op.get_attr("eid_low"), eid_high=op.get_attr("eid_high"),times = op.get_attr("times"))[1]

def e_where_equal(input1,input2,cond1,cond2):
    global eid,times