diff options
author | 2016-11-17 15:37:00 -0800 | |
---|---|---|
committer | 2016-11-17 15:45:18 -0800 | |
commit | 54e5000e0b980abe905900599c4493fadae34a15 (patch) | |
tree | c1f0751e5565882a77646589360fe090bec3f3fc /tensorflow/core/kernels/bcast_ops.cc | |
parent | 8a5610cd9f0b7087c1a7e97071ba1cf9b885315a (diff) |
Merge changes from github.
Change: 139516555
Diffstat (limited to 'tensorflow/core/kernels/bcast_ops.cc')
-rw-r--r-- | tensorflow/core/kernels/bcast_ops.cc | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/bcast_ops.cc b/tensorflow/core/kernels/bcast_ops.cc index 10354cbb56..db8842a547 100644 --- a/tensorflow/core/kernels/bcast_ops.cc +++ b/tensorflow/core/kernels/bcast_ops.cc @@ -90,4 +90,14 @@ REGISTER_KERNEL_BUILDER(Name("BroadcastGradientArgs") .HostMemory("r1"), BCastGradArgsOp); +#if TENSORFLOW_USE_SYCL +REGISTER_KERNEL_BUILDER(Name("BroadcastGradientArgs") + .Device(DEVICE_SYCL) + .TypeConstraint<int32>("T") + .HostMemory("s0") + .HostMemory("s1") + .HostMemory("r0") + .HostMemory("r1"), + BCastGradArgsOp); +#endif } // end namespace tensorflow |