Struct GpuArray 

pub struct GpuArray { /* private fields */ }

Implementations

impl GpuArray

pub fn module(&self) -> &Arc<ArrOgpuModule>

pub fn pointer(&self) -> (u32, u32)

pub fn len(&self) -> usize

pub fn shape(&self) -> &Vec<u32>

pub fn stride(&self) -> &Vec<u32>

pub fn dim(&self) -> usize

pub fn pointer_to_arr(&self) -> [u32; 2]

pub fn metadata_compound(&self) -> Option<&MetadataCompound>

impl GpuArray

pub fn get_heap(&self) -> Vec<f32>

impl GpuArray

pub fn free(self)

impl GpuArray

pub fn view(&self) -> Result<GpuArrayView<'_, GpuArray>, ArrOgpuErr>

impl GpuArray

pub fn abs(&self) -> Result<GpuArray, ArrOgpuErr>

impl GpuArray

pub fn dot<'a, A>(&'a self, arr: &'a A) -> Result<GpuArray, ArrOgpuErr>

where A: ArrayCompute + CheckArrayType<'a>,

impl GpuArray

pub fn add<A>(&self, arr: &A) -> Result<GpuArray, ArrOgpuErr>

where A: AbleElementWise,

pub fn sub<A>(&self, arr: &A) -> Result<GpuArray, ArrOgpuErr>

where A: AbleElementWise,

pub fn mul<A>(&self, arr: &A) -> Result<GpuArray, ArrOgpuErr>

where A: AbleElementWise,

pub fn div<A>(&self, arr: &A) -> Result<GpuArray, ArrOgpuErr>

where A: AbleElementWise,

impl GpuArray

pub fn log2(&self) -> Result<GpuArray, ArrOgpuErr>

impl GpuArray

pub fn matmul<A>(&self, arr: &A) -> Result<GpuArray, ArrOgpuErr>

where A: ArrayCompute,

impl GpuArray

pub fn powf<P>(&self, power: &P) -> Result<GpuArray, ArrOgpuErr>

where P: PowTrait + PowFloat,

pub fn powi<P>(&self, power: &P) -> Result<GpuArray, ArrOgpuErr>

where P: PowTrait + PowInt,

impl GpuArray

pub fn sin(&self) -> Result<GpuArray, ArrOgpuErr>

pub fn cos(&self) -> Result<GpuArray, ArrOgpuErr>

pub fn tan(&self) -> Result<GpuArray, ArrOgpuErr>

impl GpuArray

pub fn sqrt(&self) -> Result<GpuArray, ArrOgpuErr>

impl GpuArray

pub fn sum(&self) -> Result<GpuArray, ArrOgpuErr>

pub fn sum_axis(&self, axis: &[u32]) -> Result<GpuArray, ArrOgpuErr>

pub fn sum_axis_keep_dim(&self, axis: &[u32]) -> Result<GpuArray, ArrOgpuErr>

impl GpuArray

pub fn broadcast( &self, broadcast: &[u32], ) -> Result<GpuArrayView<'_, GpuArray>, ArrOgpuErr>

impl GpuArray

pub fn index( &self, index: &[i32], ) -> Result<GpuArrayView<'_, GpuArray>, ArrOgpuErr>

impl GpuArray

pub fn permute( &self, permute: &[u32], ) -> Result<GpuArrayView<'_, GpuArray>, ArrOgpuErr>

impl GpuArray

pub fn slicing( &self, slice: &[SliceRangeNegativeAble], ) -> Result<GpuArrayView<'_, GpuArray>, ArrOgpuErr>

impl GpuArray

pub fn to_shape( &self, shape: &[u32], ) -> Result<GpuArrayView<'_, GpuArray>, ArrOgpuErr>

Trait Implementations

impl ArrayCompute for GpuArray

fn module(&self) -> &Arc<ArrOgpuModule>

fn pointer_to_arr(&self) -> [u32; 2]

fn dim(&self) -> usize

fn stride(&self) -> &Vec<u32>

fn shape(&self) -> &Vec<u32>

fn pointer(&self) -> (u32, u32)

fn offset(&self) -> u32

fn len(&self) -> u32

fn is_contiguous(&self) -> bool

fn metadata_compound(&self) -> Option<&MetadataCompound>

fn check_contiguous_or_view<'a>(&'a self) -> ArrayType<'a>

impl<'a> CheckArrayType<'a> for GpuArray

fn check(&'a self) -> ArrayType<'a>

fn get(&'a self) -> &'a dyn ArrayCompute

impl Display for GpuArray

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

Formats the value using the given formatter.

impl PowTrait for GpuArray

fn get(&self) -> Result<Power, &'static str>

impl ContiguousArray for GpuArray

impl PowFloat for GpuArray