pub enum ConvBlueprint {
Forward(ForwardBlueprint),
BackwardData(BackwardDataBlueprint),
BackwardWeight(BackwardWeightBlueprint),
}Expand description
Per-operation comptime blueprint for the convolution kernel family.
The blueprint captures the minimal comptime information needed to specialize
the kernel: which operation (forward / data-grad / weight-grad), the matmul
TilingBlueprint, and per-operation comptime tunables. A different blueprint
retriggers JIT compilation, so it is kept minimal.
Variants§
Forward(ForwardBlueprint)
BackwardData(BackwardDataBlueprint)
BackwardWeight(BackwardWeightBlueprint)
Implementations§
Source§impl ConvBlueprint
impl ConvBlueprint
pub fn matmul(&self) -> &TilingBlueprint
pub fn dimensionality(&self) -> Dimensionality
Trait Implementations§
Source§impl Clone for ConvBlueprint
impl Clone for ConvBlueprint
Source§fn clone(&self) -> ConvBlueprint
fn clone(&self) -> ConvBlueprint
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for ConvBlueprint
impl Debug for ConvBlueprint
Source§impl Hash for ConvBlueprint
impl Hash for ConvBlueprint
Source§impl PartialEq for ConvBlueprint
impl PartialEq for ConvBlueprint
impl Eq for ConvBlueprint
impl StructuralPartialEq for ConvBlueprint
Auto Trait Implementations§
impl Freeze for ConvBlueprint
impl RefUnwindSafe for ConvBlueprint
impl Send for ConvBlueprint
impl Sync for ConvBlueprint
impl Unpin for ConvBlueprint
impl UnsafeUnpin for ConvBlueprint
impl UnwindSafe for ConvBlueprint
Blanket Implementations§
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<C> CloneExpand for Cwhere
C: Clone,
impl<C> CloneExpand for Cwhere
C: Clone,
fn __expand_clone_method(&self, _scope: &mut Scope) -> C
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> CompilationArg for T
impl<T> CompilationArg for T
Source§fn dynamic_cast<Arg>(&self) -> Argwhere
Arg: CompilationArg,
fn dynamic_cast<Arg>(&self) -> Argwhere
Arg: CompilationArg,
Compilation args should be the same even with different element types. However, it isn’t
possible to enforce it with the type system. So, we make the compilation args serializable
and dynamically cast them. Read more
Source§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
Source§fn equivalent(&self, key: &K) -> bool
fn equivalent(&self, key: &K) -> bool
Compare self to
key and return true if they are equal.