pub struct ConvolutionConfig<M: GlobalConfig> {
pub matmul: M,
pub params: ConvolutionParams,
}Fields§
§matmul: M§params: ConvolutionParamsImplementations§
Source§impl<M: GlobalConfig> ConvolutionConfig<M>
impl<M: GlobalConfig> ConvolutionConfig<M>
pub fn new( matmul: M, kernel_size: &[u32], stride: &[u32], dilation: &[u32], padding: &[i32], dim: Dimensionality, operation: ConvolutionOperation, ) -> Result<Self, MatmulSetupError>
Trait Implementations§
Source§impl<M: Clone + GlobalConfig> Clone for ConvolutionConfig<M>
impl<M: Clone + GlobalConfig> Clone for ConvolutionConfig<M>
Source§impl<M: GlobalConfig> ConvGemmConfig for ConvolutionConfig<M>
impl<M: GlobalConfig> ConvGemmConfig for ConvolutionConfig<M>
type GlobalMatmulConfig = M
fn matmul_config(&self) -> Self::GlobalMatmulConfig
fn vector_sizes(&self) -> MatmulVectorSizes
fn cube_dim(&self) -> CubeDim
fn check_spatial_bounds(&self) -> bool
Source§fn params(&self) -> ConvolutionParams
fn params(&self) -> ConvolutionParams
The size of the convolution kernel at
dimfn operation(&self) -> ConvolutionOperation
fn lhs_global_memory_config(&self) -> GlobalMemoryConfig
fn rhs_global_memory_config(&self) -> GlobalMemoryConfig
fn out_global_memory_config(&self) -> GlobalMemoryConfig
impl<M: Copy + GlobalConfig> Copy for ConvolutionConfig<M>
Source§impl<M: Debug + GlobalConfig> Debug for ConvolutionConfig<M>
impl<M: Debug + GlobalConfig> Debug for ConvolutionConfig<M>
Source§impl<M: GlobalConfig> Deref for ConvolutionConfig<M>
impl<M: GlobalConfig> Deref for ConvolutionConfig<M>
impl<M: Eq + GlobalConfig> Eq for ConvolutionConfig<M>
Source§impl<M: Hash + GlobalConfig> Hash for ConvolutionConfig<M>
impl<M: Hash + GlobalConfig> Hash for ConvolutionConfig<M>
Source§impl<M: PartialEq + GlobalConfig> PartialEq for ConvolutionConfig<M>
impl<M: PartialEq + GlobalConfig> PartialEq for ConvolutionConfig<M>
impl<M: PartialEq + GlobalConfig> StructuralPartialEq for ConvolutionConfig<M>
Auto Trait Implementations§
impl<M> Freeze for ConvolutionConfig<M>where
M: Freeze,
impl<M> RefUnwindSafe for ConvolutionConfig<M>where
M: RefUnwindSafe,
impl<M> Send for ConvolutionConfig<M>
impl<M> Sync for ConvolutionConfig<M>
impl<M> Unpin for ConvolutionConfig<M>where
M: Unpin,
impl<M> UnsafeUnpin for ConvolutionConfig<M>where
M: UnsafeUnpin,
impl<M> UnwindSafe for ConvolutionConfig<M>where
M: UnwindSafe,
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
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneExpand for Twhere
T: Clone,
impl<T> CloneExpand for Twhere
T: Clone,
fn __expand_clone_method(&self, _: &Scope) -> T
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
impl<T> CubeComptime for T
Source§impl<T> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere
T: Any,
Source§fn into_any(self: Box<T>) -> Box<dyn Any>
fn into_any(self: Box<T>) -> Box<dyn Any>
Converts
Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>, which can then be
downcast into Box<dyn ConcreteType> where ConcreteType implements Trait.Source§fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
Converts
Rc<Trait> (where Trait: Downcast) to Rc<Any>, which can then be further
downcast into Rc<ConcreteType> where ConcreteType implements Trait.Source§fn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
Converts
&Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &Any’s vtable from &Trait’s.Source§fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
Converts
&mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &mut Any’s vtable from &mut Trait’s.Source§impl<T> DowncastSend for T
impl<T> DowncastSend for T
Source§impl<T> DowncastSync for T
impl<T> DowncastSync for T
Source§impl<T, A> DynAccess<T> for A
impl<T, A> DynAccess<T> for A
Source§fn load(&self) -> DynGuard<T>
fn load(&self) -> DynGuard<T>
The equivalent of
Access::load.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.Source§impl<K, Q> Equivalent<Q> for K
impl<K, Q> Equivalent<Q> for K
Source§fn equivalent(&self, key: &Q) -> bool
fn equivalent(&self, key: &Q) -> bool
Compare self to
key and return true if they are equal.impl<T> ErasedDestructor for Twhere
T: 'static,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self> ⓘ
fn instrument(self, span: Span) -> Instrumented<Self> ⓘ
Source§fn in_current_span(self) -> Instrumented<Self> ⓘ
fn in_current_span(self) -> Instrumented<Self> ⓘ
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