Register quantized types for dense assign.

Change: 119104425

Register quantized types for dense assign.
Change: 119104425
51930340 · A. Unique TensorFlower · TensorFlower Gardener · 1d004f96 · 51930340
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

tensorflow/core/kernels/dense_update_ops.cc tensorflow/core/kernels/dense_update_ops.cc +1 -0

未找到文件。
--- a/tensorflow/core/kernels/dense_update_ops.cc
+++ b/tensorflow/core/kernels/dense_update_ops.cc
@@ -89,6 +89,7 @@ typedef Eigen::GpuDevice GPUDevice;
      AssignOpT<CPUDevice, type>);

 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
+TF_CALL_QUANTIZED_TYPES(REGISTER_KERNELS);
 #undef REGISTER_KERNELS

 #if GOOGLE_CUDA