diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-22 12:42:59 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-22 12:46:28 -0800 |
commit | e4532d20973c4c00854492362665317551661c18 (patch) | |
tree | 398527e29bd30d39237adb4785be5069fdb646fa /tensorflow/core/kernels/unpack_op.cc | |
parent | 673641c2d6a27fa97ee05453d671853731a4c602 (diff) |
Merge changes from github.
PiperOrigin-RevId: 179953488
Diffstat (limited to 'tensorflow/core/kernels/unpack_op.cc')
-rw-r--r-- | tensorflow/core/kernels/unpack_op.cc | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/unpack_op.cc b/tensorflow/core/kernels/unpack_op.cc index 71bbb50fb1..397bdd5670 100644 --- a/tensorflow/core/kernels/unpack_op.cc +++ b/tensorflow/core/kernels/unpack_op.cc @@ -154,6 +154,12 @@ REGISTER_KERNEL_BUILDER(Name("Unpack") .HostMemory("output") .TypeConstraint<int32>("T"), UnpackOp<CPUDevice, int32>); +REGISTER_KERNEL_BUILDER(Name("Unpack") + .Device(DEVICE_GPU) + .HostMemory("value") + .HostMemory("output") + .TypeConstraint<int64>("T"), + UnpackOp<CPUDevice, int64>); #endif // GOOGLE_CUDA @@ -171,6 +177,13 @@ REGISTER_KERNEL_BUILDER(Name("Unpack") .HostMemory("output") .TypeConstraint<int32>("T"), UnpackOp<CPUDevice, int32>); + +REGISTER_KERNEL_BUILDER(Name("Unpack") + .Device(DEVICE_SYCL) + .HostMemory("value") + .HostMemory("output") + .TypeConstraint<int64>("T"), + UnpackOp<CPUDevice, int64>); #undef REGISTER_SYCL #endif // TENSORFLOW_USE_SYCL |