Skip to main content

Tower3Batch

Type Alias Tower3Batch 

Source
pub type Tower3Batch<const K: usize> = Tower3Lane<f64x4, K>;
Expand description

The 4-rows-per-pass batched Tower3 (wide::f64x4 lanes).

Aliased Type§

pub struct Tower3Batch<const K: usize> {
    pub v: f64x4,
    pub g: [f64x4; K],
    pub h: [[f64x4; K]; K],
    pub t3: [[[f64x4; K]; K]; K],
}

Fields§

§v: f64x4

Value channel.

§g: [f64x4; K]

Gradient.

§h: [[f64x4; K]; K]

Hessian.

§t3: [[[f64x4; K]; K]; K]

Third tensor.