MCPcopy Create free account
hub / github.com/conflow-dev/ConFlow / e_where_equal

Function e_where_equal

operators.py:115–117  ·  view source on GitHub ↗
(input1,input2,cond1,cond2)

Source from the content-addressed store, hash-verified

113 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]
114
115def e_where_equal(input1,input2,cond1,cond2):
116 global eid,times
117 return trainer_ops.e_where_equal(input1,input2,cond1,cond2,eid_low=(eid&0xFFFFFFFF),eid_high=(eid>>32),times=times)
118@ops.RegisterGradient("EWhereEqual")
119def _e_where_equal_grad(op,grad):
120 with tf.name_scope("EWhereEqualGrad"), tf.xla.experimental.jit_scope(compile_ops=False):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected