Struct GroupNormRequest 

pub struct GroupNormRequest<T: CudnnSupported> {
    pub x: GpuRef<T>,
    pub y: GpuRef<T>,
    pub scale: GpuRef<T>,
    pub bias: GpuRef<T>,
    pub saved_mean: Option<GpuRef<T>>,
    pub saved_var: Option<GpuRef<T>>,
    pub dims: Vec<i64>,
    pub groups: u32,
    pub layout: TensorLayout,
    pub epsilon: f64,
    pub reply: Sender<Result<(), GpuError>>,
    pub _ty: PhantomData<T>,
}

GroupNorm — generalisation of BatchNorm/LayerNorm/InstanceNorm with groups channel partitions.

Fields

x: GpuRef<T>

y: GpuRef<T>

scale: GpuRef<T>

bias: GpuRef<T>

saved_mean: Option<GpuRef<T>>

saved_var: Option<GpuRef<T>>

dims: Vec<i64>

groups: u32

layout: TensorLayout

epsilon: f64

reply: Sender<Result<(), GpuError>>

_ty: PhantomData<T>

Trait Implementations

impl<T: CudnnSupported> CudnnDispatch for GroupNormRequest<T>

fn dtype_name(&self) -> &'static str

fn op_kind(&self) -> &'static str

fn dispatch(self: Box<Self>, _ctx: &CudnnDispatchCtx<'_>)