diff options
| author |  Brian Patton <bjp@google.com> | 2018-04-09 11:08:08 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-09 11:10:29 -0700 |
| commit | 2138a691abfa726b0b6ef28d7f3482e94ada38aa (patch) | |
| tree | 04eefcca6561f0a3d2f0b799f3636af2ca66084e /tensorflow/core/kernels/fill_functor.cu.cc | |
| parent | 9d1bf2bd4723fd3d0a012891bc54cc9db54bd9cd (diff) | |
Adds complex64/128 Fill kernel registrations for GPU.
PiperOrigin-RevId: 192153935
Diffstat (limited to 'tensorflow/core/kernels/fill_functor.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/fill_functor.cu.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/fill_functor.cu.cc b/tensorflow/core/kernels/fill_functor.cu.cc index 3487606778..050c95cf40 100644 --- a/tensorflow/core/kernels/fill_functor.cu.cc +++ b/tensorflow/core/kernels/fill_functor.cu.cc @@ -76,7 +76,7 @@ struct FillFunctor<GPUDevice, T> { }; #define DEFINE_FILL_GPU(T) template struct FillFunctor<GPUDevice, T>; -TF_CALL_REAL_NUMBER_TYPES(DEFINE_FILL_GPU); +TF_CALL_NUMBER_TYPES(DEFINE_FILL_GPU); TF_CALL_bool(DEFINE_FILL_GPU); #undef DEFINE_FILL_GPU |