external/engine/mmengine/analysis/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

from .complexity_analysis import (ActivationAnalyzer, FlopAnalyzer,
                                  activation_count, flop_count,
                                  parameter_count, parameter_count_table)
from .print_helper import get_model_complexity_info

__all__ = [
    'FlopAnalyzer', 'ActivationAnalyzer', 'flop_count', 'activation_count',
    'parameter_count', 'parameter_count_table', 'get_model_complexity_info'
]