(input1,input2,cond1,cond2)
| 121 | return trainer_ops.e_where_equal_grad(grad,op.inputs[0],op.inputs[1],op.inputs[2],op.inputs[3],eid_low=op.get_attr("eid_low"), eid_high=op.get_attr("eid_high"),times = op.get_attr("times"))[0],trainer_ops.e_where_equal_grad(grad,op.inputs[0],op.inputs[1],op.inputs[2],op.inputs[3],eid_low=op.get_attr("eid_low"), eid_high=op.get_attr("eid_high"),times = op.get_attr("times"))[1],None,None |
| 122 | |
| 123 | def e_where_gequal(input1,input2,cond1,cond2): |
| 124 | global eid,times |
| 125 | return trainer_ops.e_where_gequal(input1,input2,cond1,cond2,eid_low=(eid&0xFFFFFFFF),eid_high=(eid>>32),times=times) |
| 126 | @ops.RegisterGradient("EWhereGequal") |
| 127 | def _e_where_gequal_grad(op,grad): |
| 128 | with tf.name_scope("EWhereGequalGrad"), tf.xla.experimental.jit_scope(compile_ops=False): |
nothing calls this directly
no outgoing calls
no test coverage detected