diff options
author | A. Unique TensorFlower <nobody@tensorflow.org> | 2016-03-29 01:58:48 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-03-29 03:05:12 -0700 |
commit | 160ac73d25e30694f410c7020889536d4860689c (patch) | |
tree | 8817287dd828877462e45f82f5952cefb1b13a33 /tensorflow/core/kernels/constant_op_gpu.cu.cc | |
parent | b6d66ffda925bad1741aee0cee1e07d3546364ac (diff) |
Add half support for the first basic ops, namely Cast and Const.
Change: 118445579
Diffstat (limited to 'tensorflow/core/kernels/constant_op_gpu.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/constant_op_gpu.cu.cc | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/constant_op_gpu.cu.cc b/tensorflow/core/kernels/constant_op_gpu.cu.cc index 49159d069a..c1c56befd1 100644 --- a/tensorflow/core/kernels/constant_op_gpu.cu.cc +++ b/tensorflow/core/kernels/constant_op_gpu.cu.cc @@ -81,19 +81,17 @@ struct FillFunctor<GPUDevice, T> { #define DEFINE_FILL_GPU(T) template struct FillFunctor<GPUDevice, T> TF_CALL_REAL_NUMBER_TYPES(DEFINE_FILL_GPU); DEFINE_FILL_GPU(bool); -DEFINE_FILL_GPU(Eigen::half); #undef DEFINE_FILL_GPU // Partial specialization of FillFunctor<Device=GPUDevice, T>. template <typename T> struct SetZeroFunctor<GPUDevice, T> { void operator()(const GPUDevice& d, typename TTypes<T>::Flat out) { - To32Bit(out).device(d) = To32Bit(out).constant(T(0)); + To32Bit(out).device(d) = To32Bit(out).constant(0); } }; #define DEFINE_SETZERO_GPU(T) template struct SetZeroFunctor<GPUDevice, T> -DEFINE_SETZERO_GPU(Eigen::half); DEFINE_SETZERO_GPU(float); DEFINE_SETZERO_GPU(double); #undef DEFINE_SETZERO_GPU |