oa::FlowDenoiser

FlowDenoiser class in the OA Ml public surface.

Inheritance

public Module

Public Methods

FlowTransformer & oa::FlowDenoiser::backbone()
const FlowDenoiserConfig & oa::FlowDenoiser::config()
oa::Matrix oa::FlowDenoiser::forward(const oa::Matrix & inSample)
oa::Matrix oa::FlowDenoiser::forwardConditioned(const oa::Matrix & inSample, const oa::Matrix & inTime, const oa::Matrix & inCondition = = {}, const oa::Matrix & inTokenMask = = {})
oa::Matrix oa::FlowDenoiser::forwardGuided(const oa::Matrix & inSample, const oa::Matrix & inTime, const oa::Matrix & inCondition, oa::F32 inGuidanceScale, const oa::Matrix & inTokenMask = = {})
bool oa::FlowDenoiser::isMoe()

Constructor & Destructor Documentation

oa::FlowDenoiser::FlowDenoiser( const FlowDenoiserConfig & inConfig )
No public source comment is attached to this declaration.

Parameters

inConfig
const FlowDenoiserConfig &

Public Method Documentation

FlowTransformer & oa::FlowDenoiser::backbone()
No public source comment is attached to this declaration.

Returns

FlowTransformer &

The declared return value.

const FlowDenoiserConfig & oa::FlowDenoiser::config()
No public source comment is attached to this declaration.

Returns

const FlowDenoiserConfig &

The declared return value.

oa::Matrix oa::FlowDenoiser::forward( const oa::Matrix & inSample )
No public source comment is attached to this declaration.

Parameters

inSample
const oa::Matrix &

Returns

oa::Matrix

The declared return value.

oa::Matrix oa::FlowDenoiser::forwardConditioned( const oa::Matrix & inSample, const oa::Matrix & inTime, const oa::Matrix & inCondition = = {}, const oa::Matrix & inTokenMask = = {} )
No public source comment is attached to this declaration.

Parameters

inSample
const oa::Matrix &

inTime
const oa::Matrix &

inCondition
const oa::Matrix &

Default: = {}

inTokenMask
const oa::Matrix &

Default: = {}

Returns

oa::Matrix

The declared return value.

oa::Matrix oa::FlowDenoiser::forwardGuided( const oa::Matrix & inSample, const oa::Matrix & inTime, const oa::Matrix & inCondition, oa::F32 inGuidanceScale, const oa::Matrix & inTokenMask = = {} )
No public source comment is attached to this declaration.

Parameters

inSample
const oa::Matrix &

inTime
const oa::Matrix &

inCondition
const oa::Matrix &

inGuidanceScale
oa::F32

inTokenMask
const oa::Matrix &

Default: = {}

Returns

oa::Matrix

The declared return value.

bool oa::FlowDenoiser::isMoe()
No public source comment is attached to this declaration.

Returns

bool

The declared return value.