Home | History | Annotate | Download | only in kernels

Lines Matching refs:IdentityOp

26 REGISTER_KERNEL_BUILDER(Name("Identity").Device(DEVICE_CPU), IdentityOp);
29 REGISTER_KERNEL_BUILDER(Name("StopGradient").Device(DEVICE_CPU), IdentityOp);
32 REGISTER_KERNEL_BUILDER(Name("PreventGradient").Device(DEVICE_CPU), IdentityOp);
37 IdentityOp);
39 REGISTER_KERNEL_BUILDER(Name("RefIdentity").Device(DEVICE_CPU), IdentityOp);
44 IdentityOp);
46 IdentityOp);
52 IdentityOp); \
55 IdentityOp); \
58 IdentityOp); \
61 IdentityOp)
73 IdentityOp); \
79 IdentityOp)
91 IdentityOp); \
94 IdentityOp); \
97 IdentityOp); \
100 IdentityOp); \
104 IdentityOp)
121 IdentityOp); \
127 IdentityOp)