All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
RestrictedAttentionComponent Member List

This is the complete list of members for RestrictedAttentionComponent, including all inherited members.

Add(BaseFloat alpha, const Component &other)RestrictedAttentionComponentvirtual
Backprop(const std::string &debug_info, const ComponentPrecomputedIndexes *indexes, const CuMatrixBase< BaseFloat > &in_value, const CuMatrixBase< BaseFloat > &out_value, const CuMatrixBase< BaseFloat > &out_deriv, void *memo, Component *to_update, CuMatrixBase< BaseFloat > *in_deriv) constRestrictedAttentionComponentvirtual
BackpropOneHead(const time_height_convolution::ConvolutionComputationIo &io, const CuMatrixBase< BaseFloat > &in_value, const CuMatrixBase< BaseFloat > &c, const CuMatrixBase< BaseFloat > &out_deriv, CuMatrixBase< BaseFloat > *in_deriv) constRestrictedAttentionComponentprivate
Check() constRestrictedAttentionComponentprivate
Component()Componentinline
ConsolidateMemory()Componentinlinevirtual
context_dim_RestrictedAttentionComponentprivate
Copy() constRestrictedAttentionComponentinlinevirtual
CreateIndexesVector(const std::vector< std::pair< int32, int32 > > &n_x_pairs, int32 t_start, int32 t_step, int32 num_t_values, const std::unordered_set< Index, IndexHasher > &index_set, std::vector< Index > *output_indexes)RestrictedAttentionComponentprivatestatic
DeleteMemo(void *memo) constRestrictedAttentionComponentinlinevirtual
entropy_stats_RestrictedAttentionComponentprivate
GetComputationStructure(const std::vector< Index > &input_indexes, const std::vector< Index > &output_indexes, time_height_convolution::ConvolutionComputationIo *io) constRestrictedAttentionComponentprivate
GetIndexes(const std::vector< Index > &input_indexes, const std::vector< Index > &output_indexes, time_height_convolution::ConvolutionComputationIo &io, std::vector< Index > *new_input_indexes, std::vector< Index > *new_output_indexes) constRestrictedAttentionComponentprivate
GetInputIndexes(const MiscComputationInfo &misc_info, const Index &output_index, std::vector< Index > *desired_indexes) constRestrictedAttentionComponentvirtual
Info() constRestrictedAttentionComponentvirtual
InitFromConfig(ConfigLine *cfl)RestrictedAttentionComponentvirtual
InputDim() constRestrictedAttentionComponentinlinevirtual
IsComputable(const MiscComputationInfo &misc_info, const Index &output_index, const IndexSet &input_index_set, std::vector< Index > *used_inputs) constRestrictedAttentionComponentvirtual
key_dim_RestrictedAttentionComponentprivate
key_scale_RestrictedAttentionComponentprivate
NewComponentOfType(const std::string &type)Componentstatic
num_heads_RestrictedAttentionComponentprivate
num_left_inputs_RestrictedAttentionComponentprivate
num_left_inputs_required_RestrictedAttentionComponentprivate
num_right_inputs_RestrictedAttentionComponentprivate
num_right_inputs_required_RestrictedAttentionComponentprivate
output_context_RestrictedAttentionComponentprivate
OutputDim() constRestrictedAttentionComponentinlinevirtual
posterior_stats_RestrictedAttentionComponentprivate
PrecomputeIndexes(const MiscComputationInfo &misc_info, const std::vector< Index > &input_indexes, const std::vector< Index > &output_indexes, bool need_backprop) constRestrictedAttentionComponentvirtual
Propagate(const ComponentPrecomputedIndexes *indexes, const CuMatrixBase< BaseFloat > &in, CuMatrixBase< BaseFloat > *out) constRestrictedAttentionComponentvirtual
PropagateOneHead(const time_height_convolution::ConvolutionComputationIo &io, const CuMatrixBase< BaseFloat > &in, CuMatrixBase< BaseFloat > *c, CuMatrixBase< BaseFloat > *out) constRestrictedAttentionComponentprivate
Properties() constRestrictedAttentionComponentinlinevirtual
Read(std::istream &is, bool binary)RestrictedAttentionComponentvirtual
ReadNew(std::istream &is, bool binary)Componentstatic
ReorderIndexes(std::vector< Index > *input_indexes, std::vector< Index > *output_indexes) constRestrictedAttentionComponentvirtual
RestrictedAttentionComponent()RestrictedAttentionComponentinline
RestrictedAttentionComponent(const RestrictedAttentionComponent &other)RestrictedAttentionComponent
Scale(BaseFloat scale)RestrictedAttentionComponentvirtual
stats_count_RestrictedAttentionComponentprivate
StoreStats(const CuMatrixBase< BaseFloat > &in_value, const CuMatrixBase< BaseFloat > &out_value, void *memo)RestrictedAttentionComponentvirtual
time_stride_RestrictedAttentionComponentprivate
Type() constRestrictedAttentionComponentinlinevirtual
value_dim_RestrictedAttentionComponentprivate
Write(std::ostream &os, bool binary) constRestrictedAttentionComponentvirtual
ZeroStats()RestrictedAttentionComponentvirtual
~Component()Componentinlinevirtual