This is the complete list of members for ModelCollapser, including all inherited members.
Collapse() | ModelCollapser | inline |
CollapseComponents(int32 component_index1, int32 component_index2) | ModelCollapser | inlineprivate |
CollapseComponentsAffine(int32 component_index1, int32 component_index2) | ModelCollapser | inlineprivate |
CollapseComponentsBatchnorm(int32 component_index1, int32 component_index2) | ModelCollapser | inlineprivate |
CollapseComponentsDropout(int32 component_index1, int32 component_index2) | ModelCollapser | inlineprivate |
CollapseComponentsScale(int32 component_index1, int32 component_index2) | ModelCollapser | inlineprivate |
config_ | ModelCollapser | private |
DescriptorIsCollapsible(const Descriptor &desc) | ModelCollapser | inlineprivate |
GetDiagonallyPreModifiedComponentIndex(const CuVectorBase< BaseFloat > &offset, const CuVectorBase< BaseFloat > &scale, const std::string &src_identifier, int32 component_index) | ModelCollapser | inlineprivate |
GetScaledComponentIndex(int32 component_index, BaseFloat scale) | ModelCollapser | inlineprivate |
ModelCollapser(const CollapseModelConfig &config, Nnet *nnet) | ModelCollapser | inline |
nnet_ | ModelCollapser | private |
OptimizeNode(int32 node_index) | ModelCollapser | inlineprivate |
PreMultiplyAffineParameters(const CuVectorBase< BaseFloat > &offset, const CuVectorBase< BaseFloat > &scale, CuVectorBase< BaseFloat > *bias_params, CuMatrixBase< BaseFloat > *linear_params) | ModelCollapser | inlineprivatestatic |
ReplaceNodeInDescriptor(const Descriptor &src, int32 node_to_replace, const Descriptor &expr) | ModelCollapser | inlineprivate |
SumDescriptorIsCollapsible(const SumDescriptor &sum_desc) | ModelCollapser | inlineprivate |