Age | Commit message (Collapse) | Author |
|
* Add ArgMinMax implementation
* Add utility function to get number of elements between axis
* Add utility function to get unsigned axis
* Unit tests for ArgMinMax function
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I7bc3d610dda9526190187eb87394a8ed7a4b5cdd
|
|
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I54d2416be305db7d3fc15fb3123694ad6740b27d
|
|
Change-Id: Iba91e3f3625639f01d66f81a9f3e419e0e285d66
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
|
|
Change-Id: I7359617a307b9abb4c30b3d5f2364dc6d0f828f0
|
|
* helper methods for creating TensorShape and TensorInfo objects
Change-Id: I371fc7aea08ca6bbb9c205a143ce36e8353a1c48
|