Struct web_rwkv::context::Context

pub struct Context(/* private fields */);

Implementations

impl Context

pub fn pipeline( &self, name: &'static str ) -> Result<&ComputePipeline, TensorError>

pub fn request_shape_uniform(&self, shape: Shape) -> Arc<Buffer>

pub fn request_view_uniform(&self, view: View) -> Arc<Buffer>

impl<'a> Context

pub fn zeros<T: Scalar, Tensor: TensorInit<'a, T>>( &self, shape: Shape ) -> Tensor

pub fn ones<T: Scalar, Tensor: TensorInit<'a, T>>(&self, shape: Shape) -> Tensor

pub fn tensor_from_data<T: Scalar, Tensor: TensorInit<'a, T>>( &self, shape: Shape, data: impl Into<Cow<'a, [T]>> ) -> Result<Tensor, TensorError>

pub fn tensor_init<T: Scalar, Tensor: TensorInit<'a, T>>( &self, shape: Shape ) -> Tensor

Trait Implementations

impl Clone for Context

fn clone(&self) -> Context

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Context

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Deref for Context

type Target = Arc<ContextInner>

The resulting type after dereferencing.

fn deref(&self) -> &Self::Target

Dereferences the value.

impl DerefMut for Context

fn deref_mut(&mut self) -> &mut Self::Target

Mutably dereferences the value.

impl PartialEq for Context

fn eq(&self, other: &Self) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for Context