Struct KernelBuilder

pub struct KernelBuilder {
    pub context: Scope,
    /* private fields */
}

Prepare a kernel to create a kernel definition.

Fields

context: Scope

Cube scope.

Implementations

impl KernelBuilder

pub fn scalar(&mut self, elem: Elem) -> ExpandElement

Register a scalar and return the element to be used for kernel expansion.

pub fn output_tensor(&mut self, item: Item) -> ExpandElement

Register an output array and return the element to be used for kernel expansion.

pub fn tensor_map(&mut self) -> ExpandElement

Register a tensor map and return the element to be used for kernel expansion.

pub fn input_tensor(&mut self, item: Item) -> ExpandElement

Register an input array and return the element to be used for kernel expansion.

pub fn output_array(&mut self, item: Item) -> ExpandElement

Register an output array and return the element to be used for kernel expansion.

pub fn inplace_output(&mut self, position: Id) -> ExpandElement

Register an output that uses the same resource as the input as the given position.

pub fn input_array(&mut self, item: Item) -> ExpandElement

Register an input array and return the element to be used for kernel expansion.

pub fn build(self, settings: KernelSettings) -> KernelDefinition

Build the kernel definition.

pub fn new() -> Self

Trait Implementations

impl Default for KernelBuilder

fn default() -> Self

Returns the “default value” for a type.