(inputs,weight)
| 97 | return trainer_ops.emb_init(input_dim,output_dim,eid_low=(eid&0xFFFFFFFF),eid_high=(eid>>32)) |
| 98 | |
| 99 | def e_matmul2d(inputs,weight): |
| 100 | global eid,times |
| 101 | return trainer_ops.e_matmul2d(inputs,weight,eid_low=(eid&0xFFFFFFFF),eid_high=(eid>>32),times=times) |
| 102 | @ops.RegisterGradient("EMatmul2d") |
| 103 | def _e_matmul2d_grad(op, grad): |
| 104 | with tf.name_scope("EMatmul2dGrad"), tf.xla.experimental.jit_scope(compile_ops=False): |
nothing calls this directly
no outgoing calls
no test coverage detected