Lines Matching refs:Device
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);
36 REGISTER_KERNEL_BUILDER(Name("PlaceholderWithDefault").Device(DEVICE_CPU),
39 REGISTER_KERNEL_BUILDER(Name("RefIdentity").Device(DEVICE_CPU), IdentityOp);
43 REGISTER_KERNEL_BUILDER(Name("DebugGradientIdentity").Device(DEVICE_CPU),
45 REGISTER_KERNEL_BUILDER(Name("DebugGradientRefIdentity").Device(DEVICE_CPU),
51 Name("Identity").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
54 Name("PreventGradient").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
57 Name("RefIdentity").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
60 Name("StopGradient").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
69 .Device(DEVICE_SYCL) \
75 .Device(DEVICE_SYCL) \
90 Name("Identity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
93 Name("PreventGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
96 Name("RefIdentity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
99 Name("StopGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
102 .Device(DEVICE_GPU) \
113 // TODO(b/25387198): Also enable int32 in device memory. This kernel
117 .Device(DEVICE_GPU) \
123 .Device(DEVICE_GPU) \
