pub struct Im2col;Expand description
Im2col indexing. Loads a “column” (not the same column as im2col) of pixels into shared
memory, with a certain offset (kernel position). The corners are the bounds to load pixels
from at offset 0, so the top left corner of the kernel. The offset is added to the
corner offsets, so a (-1, -1) corner will stop the bounding box at (1, 1) for kernel
offset (2, 2).
Trait Implementations§
Source§impl CubeType for Im2col
impl CubeType for Im2col
type ExpandType = Im2colExpand
Source§fn into_mut(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType
fn into_mut(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType
Wrapper around the init method, necessary to type inference.
Source§impl LaunchArg for Im2col
impl LaunchArg for Im2col
Source§type RuntimeArg<'a, R: Runtime> = Im2colLaunch<'a, R>
type RuntimeArg<'a, R: Runtime> = Im2colLaunch<'a, R>
The runtime argument for the kernel.
Source§type CompilationArg = Im2colCompilationArg
type CompilationArg = Im2colCompilationArg
Compilation argument.
fn compilation_arg<'a, R: Runtime>( runtime_arg: &Self::RuntimeArg<'a, R>, ) -> Self::CompilationArg
Source§fn expand(
arg: &Self::CompilationArg,
builder: &mut KernelBuilder,
) -> <Self as CubeType>::ExpandType
fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType
Register an input variable during compilation that fill the KernelBuilder.
Source§fn expand_output(
arg: &Self::CompilationArg,
builder: &mut KernelBuilder,
) -> <Self as CubeType>::ExpandType
fn expand_output( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType
Register an output variable during compilation that fill the KernelBuilder.
Source§impl TensorMapKind for Im2col
impl TensorMapKind for Im2col
type Args = Im2colArgs
fn as_format(args: Self::Args) -> TensorMapFormat
impl Copy for Im2col
Auto Trait Implementations§
impl Freeze for Im2col
impl RefUnwindSafe for Im2col
impl Send for Im2col
impl Sync for Im2col
impl Unpin for Im2col
impl UnwindSafe for Im2col
Blanket Implementations§
§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
§unsafe fn clone_to_uninit(&self, dest: *mut u8)
unsafe fn clone_to_uninit(&self, dest: *mut u8)
🔬This is a nightly-only experimental API. (
clone_to_uninit)