aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/debug_ops.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/debug_ops.h')
-rw-r--r--tensorflow/core/kernels/debug_ops.h2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/debug_ops.h b/tensorflow/core/kernels/debug_ops.h
index aa47315f55..0a4dc45205 100644
--- a/tensorflow/core/kernels/debug_ops.h
+++ b/tensorflow/core/kernels/debug_ops.h
@@ -127,7 +127,7 @@ class DebugNanCountOp : public OpKernel {
const T* input_flat = input.template flat<T>().data();
for (int64 i = 0; i < input_shape.num_elements(); ++i) {
- if (Eigen::numext::isnan(input_flat[i])) {
+ if (Eigen::numext::isnan(static_cast<double>(input_flat[i]))) {
nan_count++;
}
}