Skip to main content

ClikaRT::nn::ModuleHooks

struct

Header: ClikaRT/nn/module.h

The entry points a module captures when it is constructed, one per virtual the runtime invokes from its own frames (see Module::Module() and detail::ModuleHookTable). A Result-returning hook reports a caught exception as the failed Result; the others report it as a HookOutcome. Filled by Module(); a module author never constructs one.

Data members​

output_shapes​

HookOutcome(* output_shapes)(const Module &, ClikaRT::Span< const FakeTensor >, std::vector< FakeTensor > &) noexcept = nullptr

Declared in ClikaRT/nn/module.h, line 72

compute​

HookOutcome(* compute)(const Module &, ClikaRT::Span< const Tensor >, ClikaRT::Span< Tensor >) noexcept = nullptr

Declared in ClikaRT/nn/module.h, line 74

to_where​

Result<void>(* to_where)(Module &, StreamOrDevice) noexcept = nullptr

Declared in ClikaRT/nn/module.h, line 74

to_dtype​

Result<void>(* to_dtype)(Module &, DataType) noexcept = nullptr

Declared in ClikaRT/nn/module.h, line 74

initialize​

Result<void>(* initialize)(Module &) noexcept = nullptr

Declared in ClikaRT/nn/module.h, line 74

packed_parameters​

HookOutcome(* packed_parameters)(const Module &, std::vector< std::pair< std::string, Tensor > > &) noexcept = nullptr

Declared in ClikaRT/nn/module.h, line 79

apply_weight_residency​

HookOutcome(* apply_weight_residency)(Module &, WeightResidency, std::uint64_t &, std::uint64_t &) noexcept = nullptr

Declared in ClikaRT/nn/module.h, line 81