| Add(BaseFloat alpha, const Component &other) | RestrictedAttentionComponent | virtual |
| 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) const | RestrictedAttentionComponent | virtual |
| 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) const | RestrictedAttentionComponent | private |
| Check() const | RestrictedAttentionComponent | private |
| Component() | Component | inline |
| ConsolidateMemory() | Component | inlinevirtual |
| context_dim_ | RestrictedAttentionComponent | private |
| Copy() const | RestrictedAttentionComponent | inlinevirtual |
| 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) | RestrictedAttentionComponent | privatestatic |
| DeleteMemo(void *memo) const | RestrictedAttentionComponent | inlinevirtual |
| entropy_stats_ | RestrictedAttentionComponent | private |
| GetComputationStructure(const std::vector< Index > &input_indexes, const std::vector< Index > &output_indexes, time_height_convolution::ConvolutionComputationIo *io) const | RestrictedAttentionComponent | private |
| 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) const | RestrictedAttentionComponent | private |
| GetInputIndexes(const MiscComputationInfo &misc_info, const Index &output_index, std::vector< Index > *desired_indexes) const | RestrictedAttentionComponent | virtual |
| Info() const | RestrictedAttentionComponent | virtual |
| InitFromConfig(ConfigLine *cfl) | RestrictedAttentionComponent | virtual |
| InputDim() const | RestrictedAttentionComponent | inlinevirtual |
| IsComputable(const MiscComputationInfo &misc_info, const Index &output_index, const IndexSet &input_index_set, std::vector< Index > *used_inputs) const | RestrictedAttentionComponent | virtual |
| key_dim_ | RestrictedAttentionComponent | private |
| key_scale_ | RestrictedAttentionComponent | private |
| NewComponentOfType(const std::string &type) | Component | static |
| num_heads_ | RestrictedAttentionComponent | private |
| num_left_inputs_ | RestrictedAttentionComponent | private |
| num_left_inputs_required_ | RestrictedAttentionComponent | private |
| num_right_inputs_ | RestrictedAttentionComponent | private |
| num_right_inputs_required_ | RestrictedAttentionComponent | private |
| output_context_ | RestrictedAttentionComponent | private |
| OutputDim() const | RestrictedAttentionComponent | inlinevirtual |
| posterior_stats_ | RestrictedAttentionComponent | private |
| PrecomputeIndexes(const MiscComputationInfo &misc_info, const std::vector< Index > &input_indexes, const std::vector< Index > &output_indexes, bool need_backprop) const | RestrictedAttentionComponent | virtual |
| Propagate(const ComponentPrecomputedIndexes *indexes, const CuMatrixBase< BaseFloat > &in, CuMatrixBase< BaseFloat > *out) const | RestrictedAttentionComponent | virtual |
| PropagateOneHead(const time_height_convolution::ConvolutionComputationIo &io, const CuMatrixBase< BaseFloat > &in, CuMatrixBase< BaseFloat > *c, CuMatrixBase< BaseFloat > *out) const | RestrictedAttentionComponent | private |
| Properties() const | RestrictedAttentionComponent | inlinevirtual |
| Read(std::istream &is, bool binary) | RestrictedAttentionComponent | virtual |
| ReadNew(std::istream &is, bool binary) | Component | static |
| ReorderIndexes(std::vector< Index > *input_indexes, std::vector< Index > *output_indexes) const | RestrictedAttentionComponent | virtual |
| RestrictedAttentionComponent() | RestrictedAttentionComponent | inline |
| RestrictedAttentionComponent(const RestrictedAttentionComponent &other) | RestrictedAttentionComponent | |
| Scale(BaseFloat scale) | RestrictedAttentionComponent | virtual |
| stats_count_ | RestrictedAttentionComponent | private |
| StoreStats(const CuMatrixBase< BaseFloat > &in_value, const CuMatrixBase< BaseFloat > &out_value, void *memo) | RestrictedAttentionComponent | virtual |
| time_stride_ | RestrictedAttentionComponent | private |
| Type() const | RestrictedAttentionComponent | inlinevirtual |
| value_dim_ | RestrictedAttentionComponent | private |
| Write(std::ostream &os, bool binary) const | RestrictedAttentionComponent | virtual |
| ZeroStats() | RestrictedAttentionComponent | virtual |
| ~Component() | Component | inlinevirtual |
1.8.13