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

can_cast_to​

Result<void>(*) can_cast_to(const 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

apply_weight_residency​

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

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