quark.torch.export.json_export.utils.utils

Module Contents

Functions

quark.torch.export.json_export.utils.utils.split_model_info(info: Dict[str, Any], params_dict: Dict[str, torch.Tensor]) → None

Util function to split the weights or any torch.Tensor in nested config to weights.