This is the complete list of members for BatchNormComponent, including all inherited members.
Add(BaseFloat alpha, const Component &other) | BatchNormComponent | 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 *, CuMatrixBase< BaseFloat > *in_deriv) const | BatchNormComponent | virtual |
BatchNormComponent() | BatchNormComponent | inline |
BatchNormComponent(const BatchNormComponent &other) | BatchNormComponent | |
block_dim_ | BatchNormComponent | private |
Check() const | BatchNormComponent | private |
Component() | Component | inline |
ComputeDerived() | BatchNormComponent | private |
ComputeOffsetAndScale(double count, BaseFloat epsilon, const Vector< double > &stats_sum, const Vector< double > &stats_sumsq, Vector< BaseFloat > *offset, Vector< BaseFloat > *scale) | BatchNormComponent | privatestatic |
ConsolidateMemory() | Component | inlinevirtual |
Copy() const | BatchNormComponent | inlinevirtual |
count_ | BatchNormComponent | private |
DeleteMemo(void *memo) const | BatchNormComponent | inlinevirtual |
dim_ | BatchNormComponent | private |
epsilon_ | BatchNormComponent | private |
GetInputIndexes(const MiscComputationInfo &misc_info, const Index &output_index, std::vector< Index > *desired_indexes) const | Component | virtual |
Info() const | BatchNormComponent | virtual |
InitFromConfig(ConfigLine *cfl) | BatchNormComponent | virtual |
InputDim() const | BatchNormComponent | inlinevirtual |
IsComputable(const MiscComputationInfo &misc_info, const Index &output_index, const IndexSet &input_index_set, std::vector< Index > *used_inputs) const | Component | virtual |
NewComponentOfType(const std::string &type) | Component | static |
Offset() const | BatchNormComponent | inline |
offset_ | BatchNormComponent | private |
OutputDim() const | BatchNormComponent | inlinevirtual |
PrecomputeIndexes(const MiscComputationInfo &misc_info, const std::vector< Index > &input_indexes, const std::vector< Index > &output_indexes, bool need_backprop) const | Component | inlinevirtual |
Propagate(const ComponentPrecomputedIndexes *indexes, const CuMatrixBase< BaseFloat > &in, CuMatrixBase< BaseFloat > *out) const | BatchNormComponent | virtual |
Properties() const | BatchNormComponent | inlinevirtual |
Read(std::istream &is, bool binary) | BatchNormComponent | virtual |
ReadNew(std::istream &is, bool binary) | Component | static |
ReorderIndexes(std::vector< Index > *input_indexes, std::vector< Index > *output_indexes) const | Component | inlinevirtual |
Scale(BaseFloat scale) | BatchNormComponent | virtual |
Scale() const | BatchNormComponent | inline |
scale_ | BatchNormComponent | private |
SetTestMode(bool test_mode) | BatchNormComponent | |
stats_sum_ | BatchNormComponent | private |
stats_sumsq_ | BatchNormComponent | private |
StoreStats(const CuMatrixBase< BaseFloat > &in_value, const CuMatrixBase< BaseFloat > &out_value, void *memo) | BatchNormComponent | virtual |
target_rms_ | BatchNormComponent | private |
test_mode_ | BatchNormComponent | private |
Type() const | BatchNormComponent | inlinevirtual |
Write(std::ostream &os, bool binary) const | BatchNormComponent | virtual |
ZeroStats() | BatchNormComponent | virtual |
~Component() | Component | inlinevirtual |