Trait ModuleVisitor

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>,
    ) { ... }
}

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.

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