(inputs,real)
| 105 | return trainer_ops.e_matmul2d_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_matmul2d_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] |
| 106 | |
| 107 | def e_softmax_cross_entropy(inputs,real): |
| 108 | global eid,times |
| 109 | return trainer_ops.e_softmax_cross_entropy(inputs,real,eid_low=(eid&0xFFFFFFFF),eid_high=(eid>>32),times=times) |
| 110 | @ops.RegisterGradient("ESoftmaxCrossEntropy") |
| 111 | def _e_softmax_cross_entropy_grad(op,grad): |
| 112 | with tf.name_scope("ESoftmaxCrossEntropyGrad"), tf.xla.experimental.jit_scope(compile_ops=False): |
nothing calls this directly
no outgoing calls
no test coverage detected