Function get_op_by_name

quantization/clip_utils.py:159–164

(module, op_name)

    return {name: m for name, m in module.named_modules() if isinstance(m, nn.Linear)}

def get_op_by_name(module, op_name):
    # get the op by its name relative to the module
    for name, m in module.named_modules():
        if name == op_name:
            return m
    raise ValueError(f"Cannot find op {op_name} in module {module}")

def get_op_name(module, op):
    # get the name of the op relative to the module