diff options
author | 2016-04-13 13:35:03 -0800 | |
---|---|---|
committer | 2016-04-13 14:42:11 -0700 | |
commit | 53b9ba7c821ad1a44b63be58a93d3cd1859aac3d (patch) | |
tree | dca142f13e15476ee2c24c72d419cca37376e652 /tensorflow/core/kernels/resize_area_op.cc | |
parent | 021f88578a15986b8d5b3cc2db0291fc3ab4dde4 (diff) |
Make TF_CALL_REAL_NUMBER_TYPES() and related macros include Eigen::half
in addition to float. Explicitly exempt GPU effects that rely on atomics,
for which we have no good solution for half yet.
Note that there are still ops that don't _declare_ half support (those
that use \u201callnumbertypes\u201d or similar do, those that use \u201cfloat, double\u201d
don't); these will be fixed in a forthcoming commit.
Change: 119786433
Diffstat (limited to 'tensorflow/core/kernels/resize_area_op.cc')
-rw-r--r-- | tensorflow/core/kernels/resize_area_op.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/resize_area_op.cc b/tensorflow/core/kernels/resize_area_op.cc index 5bf064f159..032db0cb5e 100644 --- a/tensorflow/core/kernels/resize_area_op.cc +++ b/tensorflow/core/kernels/resize_area_op.cc @@ -103,8 +103,8 @@ class ResizeAreaOp : public OpKernel { j < in_x ? j + 1 - in_x : (j + 1 > in_x1 ? in_x1 - j : 1.0); for (int64 c = 0; c < st.channels; ++c) { #define BOUND(val, limit) std::min(((limit)-1ll), (std::max(0ll, (val)))) - sum_data(c) += input_data(b, BOUND(i, st.in_height), - BOUND(j, st.in_width), c) * + sum_data(c) += float(input_data(b, BOUND(i, st.in_height), + BOUND(j, st.in_width), c)) * scale_y * scale_x * scale; #undef BOUND } |