Commit 1d0552e9 authored by Alan Chiao's avatar Alan Chiao Committed by TensorFlower Gardener
Browse files

Match TFLite argmin/argmax shape propagation to Tensorflow. 

- output shape is now input shape - 1
- handle axis values other than axis = 3

PiperOrigin-RevId: 230450280
parent 25d117af
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