Trait burn::module::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.

Object Safety

This trait is not object safe.

Implementors

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

where B: Backend, C: Calibration,