Struct GpuOps 

pub struct GpuOps { /* private fields */ }

GPU operations dispatcher with cached pipelines.

Implementations

impl GpuOps

pub fn new(device: Arc<GpuDevice>) -> Self

pub fn add(&self, a: &GpuTensor, b: &GpuTensor) -> Result<GpuTensor, String>

pub fn sub(&self, a: &GpuTensor, b: &GpuTensor) -> Result<GpuTensor, String>

pub fn mul(&self, a: &GpuTensor, b: &GpuTensor) -> Result<GpuTensor, String>

pub fn div(&self, a: &GpuTensor, b: &GpuTensor) -> Result<GpuTensor, String>

pub fn scale(&self, a: &GpuTensor, scalar: f32) -> GpuTensor

pub fn sum(&self, a: &GpuTensor) -> Result<f32, String>

pub fn mean(&self, a: &GpuTensor) -> Result<f32, String>

pub fn matmul(&self, a: &GpuTensor, b: &GpuTensor) -> Result<GpuTensor, String>

pub fn transpose(&self, a: &GpuTensor) -> Result<GpuTensor, String>