diff options
| author |  Jianmin Chen <jmchen@google.com> | 2016-10-14 15:54:21 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-10-14 17:05:45 -0700 |
| commit | 5d1c0c41510f51a707373f0d1efabf3be44fa480 (patch) | |
| tree | 17ceb20bcf6f5a49024c8270bc3a94d2bf00749f /tensorflow/core/kernels/constant_op.cc | |
| parent | 8ce0576beb536446473abf8d6562c60c7ff838d7 (diff) | |
Register Fill with quint8.
Change: 136214096
Diffstat (limited to 'tensorflow/core/kernels/constant_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/constant_op.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/constant_op.cc b/tensorflow/core/kernels/constant_op.cc index be7a7a41a4..97608938a4 100644 --- a/tensorflow/core/kernels/constant_op.cc +++ b/tensorflow/core/kernels/constant_op.cc @@ -155,6 +155,7 @@ class FillOp : public OpKernel { #define REGISTER_CPU_KERNEL(TYPE) REGISTER_KERNEL(CPU, TYPE) TF_CALL_ALL_TYPES(REGISTER_CPU_KERNEL); +REGISTER_KERNEL(CPU, quint8); #undef REGISTER_CPU_KERNEL #if GOOGLE_CUDA |