Commit 94b19f2a authored by Yong Tang's avatar Yong Tang
Browse files

Use FastBoundsCheck in ArgMax kernel op 



This fix updates ArgMax kernel implementation to
use FastBoundsCheck for improved performance, and
keep consistency with other places in tf.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent fe7d1d94
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