Enum BatchableOp 

pub enum BatchableOp<T: TensorElement> {
    Add(Arc<Tensor<T>>, Arc<Tensor<T>>),
    Mul(Arc<Tensor<T>>, Arc<Tensor<T>>),
    Sub(Arc<Tensor<T>>, Arc<Tensor<T>>),
    Div(Arc<Tensor<T>>, Arc<Tensor<T>>),
    AddScalar(Arc<Tensor<T>>, T),
    MulScalar(Arc<Tensor<T>>, T),
    ReLU(Arc<Tensor<T>>),
    Sigmoid(Arc<Tensor<T>>),
    Tanh(Arc<Tensor<T>>),
}

Type of tensor operation that can be batched

Variants

Add(Arc<Tensor<T>>, Arc<Tensor<T>>)

Element-wise addition

Mul(Arc<Tensor<T>>, Arc<Tensor<T>>)

Element-wise multiplication

Sub(Arc<Tensor<T>>, Arc<Tensor<T>>)

Element-wise subtraction

Div(Arc<Tensor<T>>, Arc<Tensor<T>>)

Element-wise division

AddScalar(Arc<Tensor<T>>, T)

Scalar addition

MulScalar(Arc<Tensor<T>>, T)

Scalar multiplication

ReLU(Arc<Tensor<T>>)

ReLU activation

Sigmoid(Arc<Tensor<T>>)

Sigmoid activation

Tanh(Arc<Tensor<T>>)

Tanh activation

Implementations

impl<T: TensorElement> BatchableOp<T>

pub fn size(&self) -> usize

Get the estimated size (in elements) of the operation

pub fn device(&self) -> DeviceType

Get the device type of the operation

pub fn should_batch(&self, config: &BatchingConfig) -> bool

Check if this operation should be batched based on config

Trait Implementations

impl<T: Clone + TensorElement> Clone for BatchableOp<T>

fn clone(&self) -> BatchableOp<T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: Debug + TensorElement> Debug for BatchableOp<T>

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

Formats the value using the given formatter.