#[repr(C)]pub struct ReduceParams {
pub outer: u32,
pub reduce_dim: u32,
pub inner: u32,
pub in_off: u32,
pub out_off: u32,
pub op: u32,
pub _p0: u32,
pub _p1: u32,
}Expand description
Layout for reductions. 32 bytes.
Supports arbitrary-axis reductions. The reduce kernel walks the
input as a 3D tensor [outer, reduce_dim, inner] where:
outer= product of dims BEFORE the reduce axisreduce_dim= the reduce axis itselfinner= product of dims AFTER the reduce axis (=1 for the last-axis case, which is what the v3 dispatcher emitted). Output shape is[outer, inner](or with the reduce axis kept as 1 whenkeep_dim; the dispatcher handles the shape arithmetic).
Fields§
§outer: u32§reduce_dim: u32§inner: u32§in_off: u32§out_off: u32§op: u32§_p0: u32§_p1: u32Trait Implementations§
Source§impl Clone for ReduceParams
impl Clone for ReduceParams
impl Copy for ReduceParams
Source§impl Debug for ReduceParams
impl Debug for ReduceParams
impl Pod for ReduceParams
Auto Trait Implementations§
impl Freeze for ReduceParams
impl RefUnwindSafe for ReduceParams
impl Send for ReduceParams
impl Sync for ReduceParams
impl Unpin for ReduceParams
impl UnsafeUnpin for ReduceParams
impl UnwindSafe for ReduceParams
Blanket Implementations§
impl<T> AnyBitPattern for Twhere
T: Pod,
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CheckedBitPattern for Twhere
T: AnyBitPattern,
impl<T> CheckedBitPattern for Twhere
T: AnyBitPattern,
Source§type Bits = T
type Bits = T
Self must have the same layout as the specified Bits except for
the possible invalid bit patterns being checked during
is_valid_bit_pattern.Source§fn is_valid_bit_pattern(_bits: &T) -> bool
fn is_valid_bit_pattern(_bits: &T) -> bool
If this function returns true, then it must be valid to reinterpret
bits
as &Self.Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more