ModelCollapser Member List

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

Collapse()ModelCollapserinline
CollapseComponents(int32 component_index1, int32 component_index2)ModelCollapserinlineprivate
CollapseComponentsAffine(int32 component_index1, int32 component_index2)ModelCollapserinlineprivate
CollapseComponentsBatchnorm(int32 component_index1, int32 component_index2)ModelCollapserinlineprivate
CollapseComponentsDropout(int32 component_index1, int32 component_index2)ModelCollapserinlineprivate
CollapseComponentsScale(int32 component_index1, int32 component_index2)ModelCollapserinlineprivate
config_ModelCollapserprivate
DescriptorIsCollapsible(const Descriptor &desc)ModelCollapserinlineprivate
GetDiagonallyPreModifiedComponentIndex(const CuVectorBase< BaseFloat > &offset, const CuVectorBase< BaseFloat > &scale, const std::string &src_identifier, int32 component_index)ModelCollapserinlineprivate
GetScaledComponentIndex(int32 component_index, BaseFloat scale)ModelCollapserinlineprivate
ModelCollapser(const CollapseModelConfig &config, Nnet *nnet)ModelCollapserinline
nnet_ModelCollapserprivate
OptimizeNode(int32 node_index)ModelCollapserinlineprivate
PreMultiplyAffineParameters(const CuVectorBase< BaseFloat > &offset, const CuVectorBase< BaseFloat > &scale, CuVectorBase< BaseFloat > *bias_params, CuMatrixBase< BaseFloat > *linear_params)ModelCollapserinlineprivatestatic
ReplaceNodeInDescriptor(const Descriptor &src, int32 node_to_replace, const Descriptor &expr)ModelCollapserinlineprivate
SumDescriptorIsCollapsible(const SumDescriptor &sum_desc)ModelCollapserinlineprivate