Trait ModuleMapper

pub trait ModuleMapper<B>
where
    B: Backend,
{
    // Provided methods
    fn map_float<const D: usize>(
        &mut self,
        _id: ParamId,
        tensor: Tensor<B, D>,
    ) -> Tensor<B, D> { ... }
    fn map_int<const D: usize>(
        &mut self,
        _id: ParamId,
        tensor: Tensor<B, D, Int>,
    ) -> Tensor<B, D, Int> { ... }
    fn map_bool<const D: usize>(
        &mut self,
        _id: ParamId,
        tensor: Tensor<B, D, Bool>,
    ) -> Tensor<B, D, Bool> { ... }
}

Module mapper trait.

Provided Methods

fn map_float<const D: usize>( &mut self, _id: ParamId, tensor: Tensor<B, D>, ) -> Tensor<B, D>

Map a float tensor in the module.

fn map_int<const D: usize>( &mut self, _id: ParamId, tensor: Tensor<B, D, Int>, ) -> Tensor<B, D, Int>

Map an int tensor in the module.

fn map_bool<const D: usize>( &mut self, _id: ParamId, tensor: Tensor<B, D, Bool>, ) -> Tensor<B, D, Bool>

Map a bool tensor in the module.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<B, C> ModuleMapper<B> for Quantizer<C>

where B: Backend, C: Calibration,