pub struct Lppool3dBackward<D: Float> {
pub name: &'static str,
pub id: String,
pub kd: i32,
pub kh: i32,
pub kw: i32,
pub stride_d: i32,
pub stride_h: i32,
pub stride_w: i32,
pub block_ow: i32,
pub kernel_source: String,
pub entry_point_source: String,
pub source: String,
/* private fields */
}Expand description
3-D Lp-norm pooling backward pass.
dx_i = dy * sign(x_i) * (|x_i| / max(y, ε))^(p-1) / max(y, ε).
dx must be zero-initialised before launch.
Fields§
§name: &'static strThe kernel function’s name (e.g. "flash_attention2_forward").
id: StringUnique kernel identifier: fn_name + dtype(s) + const values joined by “__”.
kd: i32Compile-time kernel constant, from the annotated fn’s const generics.
kh: i32Compile-time kernel constant, from the annotated fn’s const generics.
kw: i32Compile-time kernel constant, from the annotated fn’s const generics.
stride_d: i32Compile-time kernel constant, from the annotated fn’s const generics.
stride_h: i32Compile-time kernel constant, from the annotated fn’s const generics.
stride_w: i32Compile-time kernel constant, from the annotated fn’s const generics.
block_ow: i32Compile-time kernel constant, from the annotated fn’s const generics.
kernel_source: StringThe original kernel function source.
entry_point_source: StringThe Rust source of the generated C-ABI entry-point wrapper function.
source: StringCombined source (kernel_source + "\n\n" + entry_point_source); used by the Kernel trait.
Implementations§
Trait Implementations§
Source§impl<D: Float> Kernel for Lppool3dBackward<D>
impl<D: Float> Kernel for Lppool3dBackward<D>
Source§type Args<'__a> = (*mut D, *mut D, *mut D, *mut D, i32, i32, i32, i32, i32, i32, i32, i32, f32)
type Args<'__a> = (*mut D, *mut D, *mut D, *mut D, i32, i32, i32, i32, i32, i32, i32, i32, f32)
Source§fn kernel_source(&self) -> &str
fn kernel_source(&self) -> &str
Source§fn entry_point_source(&self) -> &str
fn entry_point_source(&self) -> &str
Source§fn entry_point_name(&self) -> String
fn entry_point_name(&self) -> String
"{name}_entry_point".