.. _f-onnxsplit: module ``utils.onnx_split`` =========================== .. inheritance-diagram:: onnxcustom.utils.onnx_split Short summary +++++++++++++ module ``onnxcustom.utils.onnx_split`` Helpers to split an ONNX models. :githublink:`%|py|5` Classes +++++++ +--------------------------------------------------------------------+-------------------------------------------------------------------------------+ | class | truncated documentation | +====================================================================+===============================================================================+ | :class:`OnnxSegment ` | A segments of an onnx graph assuming it is the concatenation of all segments. | +--------------------------------------------------------------------+-------------------------------------------------------------------------------+ | :class:`OnnxSplitting ` | The final goal is to split an onnx model into equivalent pieces. | +--------------------------------------------------------------------+-------------------------------------------------------------------------------+ Functions +++++++++ +-------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------------+ | function | truncated documentation | +=============================================================+============================================================================================================================+ | :func:`split_onnx ` | Splits an ONNX model into *n_parts* consecutive subgraphs. Chained altogether, they are equivalent to the given model. ... | +-------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------------+ Static Methods ++++++++++++++ +----------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------------------------------+ | staticmethod | truncated documentation | +==============================================================================================+=======================================================================================================================+ | :py:meth:`_connex_components ` | | +----------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------------------------------+ | :py:meth:`_key ` | | +----------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------------------------------+ | :meth:`is_small ` | Tells if a tensor is small. In that case, all edges to this constant are ignored when looking for cutting points. ... | +----------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------------------------------+ Methods +++++++ +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | method | truncated documentation | +================================================================================================+========================================================================+ | :py:meth:`__init__ ` | | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | :py:meth:`__init__ ` | | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | :py:meth:`__repr__ ` | | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | :py:meth:`_get_cutting_points ` | | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | :py:meth:`_init ` | | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | :py:meth:`_make_onnx ` | Builds one onnx subpart including segments from a to b (excluded). | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | :py:meth:`_make_segment ` | | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | :py:meth:`_split_2 ` | Splits the segments into two groups of the same size. | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | :meth:`make_onnx ` | Builds onnx subparts based on the segmentation defined by extremities. | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ | :meth:`split_segment ` | Splits the segments into `n_parts` segments | +------------------------------------------------------------------------------------------------+------------------------------------------------------------------------+ Documentation +++++++++++++ .. automodule:: onnxcustom.utils.onnx_split :members: :special-members: __init__ :show-inheritance: