Searched refs:torchao_optimize_ctx (Results 1 – 2 of 2) sorted by relevance
14 def torchao_optimize_ctx(quantization: str): function
3980 from torchao_backend import setup_baseline, torchao_optimize_ctx3984 torchao_optimize_ctx,3995 optimize_ctx = torchao_optimize_ctx(args.quantization)