#include <nnet-attention-component.h>
Public Member Functions | |
PrecomputedIndexes () | |
PrecomputedIndexes (const PrecomputedIndexes &other) | |
virtual PrecomputedIndexes * | Copy () const |
virtual void | Write (std::ostream &os, bool binary) const |
virtual void | Read (std::istream &os, bool binary) |
virtual std::string | Type () const |
virtual | ~PrecomputedIndexes () |
Public Member Functions inherited from ComponentPrecomputedIndexes | |
virtual | ~ComponentPrecomputedIndexes () |
Public Attributes | |
time_height_convolution::ConvolutionComputationIo | io |
Additional Inherited Members | |
Static Public Member Functions inherited from ComponentPrecomputedIndexes | |
static ComponentPrecomputedIndexes * | ReadNew (std::istream &is, bool binary) |
static ComponentPrecomputedIndexes * | NewComponentPrecomputedIndexesOfType (const std::string &cpi_type) |
Definition at line 185 of file nnet-attention-component.h.
|
inline |
Definition at line 187 of file nnet-attention-component.h.
|
inline |
Definition at line 188 of file nnet-attention-component.h.
References RestrictedAttentionComponent::PrecomputedIndexes::Copy(), RestrictedAttentionComponent::PrecomputedIndexes::Read(), and RestrictedAttentionComponent::PrecomputedIndexes::Write().
|
inlinevirtual |
Definition at line 196 of file nnet-attention-component.h.
|
virtual |
Implements ComponentPrecomputedIndexes.
Definition at line 649 of file nnet-attention-component.cc.
Referenced by RestrictedAttentionComponent::PrecomputedIndexes::PrecomputedIndexes().
|
virtual |
Implements ComponentPrecomputedIndexes.
Definition at line 661 of file nnet-attention-component.cc.
References kaldi::ExpectOneOrTwoTokens(), and kaldi::nnet3::ExpectToken().
Referenced by RestrictedAttentionComponent::PrecomputedIndexes::PrecomputedIndexes().
|
inlinevirtual |
Implements ComponentPrecomputedIndexes.
Definition at line 193 of file nnet-attention-component.h.
|
virtual |
Implements ComponentPrecomputedIndexes.
Definition at line 653 of file nnet-attention-component.cc.
References kaldi::WriteToken().
Referenced by RestrictedAttentionComponent::PrecomputedIndexes::PrecomputedIndexes().
Definition at line 198 of file nnet-attention-component.h.
Referenced by RestrictedAttentionComponent::Backprop(), RestrictedAttentionComponent::PrecomputeIndexes(), and RestrictedAttentionComponent::Propagate().