Int32 support for Empty op on GPU.

PiperOrigin-RevId: 236790821
tensorflow · Mar 5, 2019 · 9707af5 · 9707af5
1 parent b71b8ec
commit 9707af5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/core/kernels/inplace_ops.cc b/tensorflow/core/kernels/inplace_ops.cc
@@ -543,6 +543,7 @@ REGISTER_EMPTY(float, GPU);
 REGISTER_EMPTY(double, GPU);
 REGISTER_EMPTY(Eigen::half, GPU);
 REGISTER_EMPTY(int64, GPU);
+REGISTER_EMPTY(int32, GPU);
 
 #endif  // GOOGLE_CUDA