OpArgMaxAPI

Trait OpArgMaxAPI 

Source
pub trait OpArgMaxAPI<T, D>
where D: DimAPI, Self: DeviceAPI<T> + DeviceAPI<Self::TOut>,
{ type TOut; // Required methods fn argmax_all( &self, a: &<Self as DeviceRawAPI<T>>::Raw, la: &Layout<D>, ) -> Result<Self::TOut>; fn argmax_axes( &self, a: &<Self as DeviceRawAPI<T>>::Raw, la: &Layout<D>, axes: &[isize], ) -> Result<(Storage<DataOwned<<Self as DeviceRawAPI<Self::TOut>>::Raw>, Self::TOut, Self>, Layout<IxD>)>; }

Required Associated Types§

Required Methods§

Source

fn argmax_all( &self, a: &<Self as DeviceRawAPI<T>>::Raw, la: &Layout<D>, ) -> Result<Self::TOut>

Source

fn argmax_axes( &self, a: &<Self as DeviceRawAPI<T>>::Raw, la: &Layout<D>, axes: &[isize], ) -> Result<(Storage<DataOwned<<Self as DeviceRawAPI<Self::TOut>>::Raw>, Self::TOut, Self>, Layout<IxD>)>

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§

Source§

impl<T, D> OpArgMaxAPI<T, D> for DeviceCpuSerial
where T: Clone + PartialOrd, D: DimAPI,

Source§

impl<T, D> OpArgMaxAPI<T, D> for DeviceFaer
where T: Clone + PartialOrd + Send + Sync, D: DimAPI,