projects / platform / upstream / tensorflow.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 08b68c8)
Register GPU host kernels for Identity and RefIdentity.
authorA. Unique TensorFlower <gardener@tensorflow.org>
Fri, 2 Feb 2018 18:58:12 +0000 (10:58 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 2 Feb 2018 19:15:36 +0000 (11:15 -0800)
PiperOrigin-RevId: 184305574

tensorflow/core/kernels/identity_op.cc patch | blob | history

diff --git a/tensorflow/core/kernels/identity_op.cc b/tensorflow/core/kernels/identity_op.cc
index 1db9263..a18a72c 100644 (file)
--- a/tensorflow/core/kernels/identity_op.cc
+++ b/tensorflow/core/kernels/identity_op.cc
@@ -128,6 +128,7 @@ REGISTER_GPU_KERNEL(Variant);
 
 REGISTER_GPU_HOST_KERNEL(int32);
 REGISTER_GPU_HOST_KERNEL(bool);
+REGISTER_GPU_HOST_KERNEL(string);
 
 #undef REGISTER_GPU_HOST_KERNEL
 
Domain: Machine Learning / ML Framework;