diff options
| author |  Yan Facai (颜发才) <facai.yan@gmail.com> | 2018-08-22 11:07:05 +0800 |
|---|---|---|
| committer | Yan Facai (颜发才) <facai.yan@gmail.com> | 2018-08-22 11:57:19 +0800 |
| commit | 05a35adbe4456e6cf7854dcb4b2a113c6c05fd24 (patch) | |
| tree | ea01b6075af4a7d98e60e70b76a394129d316143 /tensorflow/core/kernels/cwise_op_div.cc | |
| parent | b18cbef5772abd61d841b2b765f3a203f8616f2e (diff) | |
ENH: register gpu kernel for div_no_nan
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_div.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_div.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_op_div.cc b/tensorflow/core/kernels/cwise_op_div.cc index 35662e278f..313d976e2c 100644 --- a/tensorflow/core/kernels/cwise_op_div.cc +++ b/tensorflow/core/kernels/cwise_op_div.cc @@ -33,6 +33,7 @@ REGISTER4(BinaryOp, GPU, "TruncateDiv", functor::div, uint8, uint16, int16, int64); REGISTER5(BinaryOp, GPU, "RealDiv", functor::div, float, Eigen::half, double, complex64, complex128); +REGISTER2(BinaryOp, GPU, "DivNoNan", functor::div_no_nan, float, double); // A special GPU kernel for int32. // TODO(b/25387198): Also enable int32 in device memory. This kernel |