TrainingExtensions/torch/src/python/aimet_torch/meta/connectedgraph.py
def _create_split_op(self, op: Op) -> Op:
        """
        The op's output is split in the forward function. To model it correctly, create a Split Op.
        :param op: Op to create split op after
        :return: Split op that was created
        """
        split_name_parts = ['Split_', str(self._split_count)]
        split_name = ''.join(split_name_parts)
        self._split_count += 1
        split_dotted_name_parts = [self._model_name, split_name]
        split_dotted_name = '.'.join(split_dotted_name_parts)
        is_anonymous = True
        split_op = Op(name=split_name, dotted_name=split_dotted_name, output_shape=op.output_shape,
                      is_anonymous=is_anonymous, op_type='Split')
        self._ops[split_name] = split_op
        return split_op

        The op's output is split in the forward function. To model it correctly, create a Split Op.
        :param op: Op to create split op after
        :return: Split op that was created
        
The op's output is split in the forward function. To model it correctly, create a Split Op.
The op ' s output is split in the forward function . To model it correctly create a Split Op .
def _create_split_op(self, op: Op) -> Op:
        split_name_parts = ['Split_', str(self._split_count)]
        split_name = ''.join(split_name_parts)
        self._split_count += 1
        split_dotted_name_parts = [self._model_name, split_name]
        split_dotted_name = '.'.join(split_dotted_name_parts)
        is_anonymous = True
        split_op = Op(name=split_name, dotted_name=split_dotted_name, output_shape=op.output_shape,
                      is_anonymous=is_anonymous, op_type='Split')
        self._ops[split_name] = split_op
        return split_op
def _create_split_op ( self , op : Op ) -> Op : split_name_parts = [ 'Split_' , str ( self . _split_count ) ] split_name = '' . join ( split_name_parts ) self . _split_count += 1 split_dotted_name_parts = [ self . _model_name , split_name ] split_dotted_name = '.' . join ( split_dotted_name_parts ) is_anonymous = True split_op = Op ( name = split_name , dotted_name = split_dotted_name , output_shape = op . output_shape , is_anonymous = is_anonymous , op_type = 'Split' ) self . _ops [ split_name ] = split_op return split_op
The op's output is split in the forward function.
The op ' s output is split in the forward function .
"""
        The op's output is split in the forward function. To model it correctly, create a Split Op.
        :param op: Op to create split op after
        :return: Split op that was created
        """
Split op that was created
Split op that was created
self
op
Op
Op to create split op after
Op to create split op after