Trait burn_core::module::ModuleVisitor

pub trait ModuleVisitor<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>
    ) { ... }
}

Module visitor trait.

Provided Methods

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

Visit a float tensor in the module.

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

Visit an int tensor in the module.

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