def collect_extenders(path: str): """ A function to registrate all MO IR Reader extenders :param path: Path to Model Optimizer folder :return: """ import_by_path(os.path.join(path, 'mo', 'utils', 'ir_reader', 'extenders'), ['mo', 'utils', 'ir_reader', 'extenders']) update_registration(classes=[Extender], enabled_transforms=[], disabled_transforms=[])
def collect_ops(path: str): """ A function to registrate all MO ops :param path: Path to Model Optimizer folder :return: """ import_by_path(os.path.join(path, 'mo', 'ops'), ['mo', 'ops']) import_by_path(os.path.join(path, 'extensions', 'ops'), ['extensions', 'ops']) update_registration(classes=[Op, Activation, Elementwise, LogicalElementwise, ReduceOp, Scatter], enabled_transforms=[], disabled_transforms=[])