diff options
author | A. Unique TensorFlower <nobody@tensorflow.org> | 2016-04-15 04:56:05 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-04-15 06:02:54 -0700 |
commit | 31fd4868711f393bec74200231c2936bf3df079a (patch) | |
tree | 6cce655cc9690f232dbfd0725c7175e8d333f8bd /tensorflow/core/kernels/cwise_op_gpu_ceil.cu.cc | |
parent | df15baa9b10a0b2d194181dff7ee14bff70d9b8f (diff) |
fp16-enable all the componentwise ops.
This also includes updating parts of the Python test framework to handle fp16.
fp16 is too inaccurate to do numerical gradients unless a lot of care is taken,
so for fp16, we compare fp16 theoretical gradients to fp32 numerical ones.
This means that the gradient check doesn't also implicitly test the function
itself, so we will need to rely on the numpy tests for those.
Change: 119948035
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_gpu_ceil.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/cwise_op_gpu_ceil.cu.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cwise_op_gpu_ceil.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_ceil.cu.cc index 760f5ad5b6..ddff6e7c78 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_ceil.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_ceil.cu.cc @@ -19,7 +19,7 @@ limitations under the License. namespace tensorflow { namespace functor { -DEFINE_UNARY2(ceil, float, double); +DEFINE_UNARY3(ceil, Eigen::half, float, double); } // namespace functor } // namespace tensorflow |