TrainingExtensions/torch/src/python/aimet_torch/meta/connectedgraph.py
def _parse_op_type(node: torch._C.Node) -> str:
        """
        Helper method to extract op type from node info
        :param node: trace graph node
        :return: Op Type string
        """
        # extracting Op type from node.kind string e.g. aten::relu_, aten::size etc
        op_type = node.kind().split("::")[-1].lstrip('_').rstrip('_')
        return op_type

        Helper method to extract op type from node info
        :param node: trace graph node
        :return: Op Type string
        
Helper method to extract op type from node info
Helper method to extract op type from node info
def _parse_op_type(node: torch._C.Node) -> str:
        op_type = node.kind().split("::")[-1].lstrip('_').rstrip('_')
        return op_type
def _parse_op_type ( node : torch . _C . Node ) -> str : op_type = node . kind ( ) . split ( "::" ) [ - 1 ] . lstrip ( '_' ) . rstrip ( '_' ) return op_type
Helper method to extract op type from node info
Helper method to extract op type from node info
"""
        Helper method to extract op type from node info
        :param node: trace graph node
        :return: Op Type string
        """ # extracting Op type from node.kind string e.g. aten::relu_, aten::size etc
Op Type string
Op Type string
node
torch._C.Node
trace graph node
trace graph node