Trait burn::module::ModuleVisitor

source · 
pub trait ModuleVisitor<B>
where
    B: Backend,
{
    // Provided methods
    fn visit_float<const D: usize>(
        &mut self,
        _id: &ParamId,
        _tensor: &Tensor<B, D>
    ) { ... }
    fn visit_int<const D: usize>(
        &mut self,
        _id: &ParamId,
        _tensor: &Tensor<B, D, Int>
    ) { ... }
    fn visit_bool<const D: usize>(
        &mut self,
        _id: &ParamId,
        _tensor: &Tensor<B, D, Bool>
    ) { ... }
}
Expand description

Module visitor trait.

Provided Methods§

source

fn visit_float<const D: usize>(&mut self, _id: &ParamId, _tensor: &Tensor<B, D>)

Visit a float tensor in the module.

source

fn visit_int<const D: usize>( &mut self, _id: &ParamId, _tensor: &Tensor<B, D, Int> )

Visit an int tensor in the module.

source

fn visit_bool<const D: usize>( &mut self, _id: &ParamId, _tensor: &Tensor<B, D, Bool> )

Visit a bool tensor in the module.

Object Safety§

This trait is not object safe.

Implementors§