Skip to main content

ReduceDTypeKind

luma_tensor::ops

Trait ReduceDTypeKind 

Source 
pub trait ReduceDTypeKind<D: Device>: DTypeKind<D> {
    // Required methods
    fn arg_reduce_dispatch(
        x: &Self::Storage,
        layout: &Layout,
        dim: usize,
        keepdim: bool,
        take_max: bool,
    ) -> Result<(D::IntStorage, Shape)>;
    fn reduce_dispatch(
        x: &Self::Storage,
        l: &Layout,
        dims: &[usize],
        keepdim: bool,
        op: ReduceOp,
    ) -> Result<(Self::Storage, Shape)>;
}

Required Methods§

Source

fn arg_reduce_dispatch( x: &Self::Storage, layout: &Layout, dim: usize, keepdim: bool, take_max: bool, ) -> Result<(D::IntStorage, Shape)>

Source

fn reduce_dispatch( x: &Self::Storage, l: &Layout, dims: &[usize], keepdim: bool, op: ReduceOp, ) -> Result<(Self::Storage, Shape)>

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors§