Trait burn_core::module::ModuleMapper

pub trait ModuleMapper<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: Backend, C: Calibration> ModuleMapper<B> for Quantizer<C>