Commit f8b35e00 authored by Adrian Kuegel's avatar Adrian Kuegel Committed by TensorFlower Gardener
Browse files

Choose the correct output type for ResizeNearestNeighbour. 

For ResizeNearestNeighbour the output type should be the same as the input
type. So far, we always chose output type F32, which is correct for
ResizeBilinear, but not for ResizeNearestNeighbour.
Finally replace Adds which are just broadcasts by BroadcastInDim.

PiperOrigin-RevId: 234116435
parent 5d6f896e
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment