（beta）torch_npu.jit.optimize

接口原型

torch_npu.jit.optimize(jit_mod)

功能描述

实现jit_mod优化。

参数说明

jit_mod：用于被优化的ScriptFunction or ScriptModule。

支持的型号

• Atlas 训练系列产品
• Atlas A2 训练系列产品
• Atlas 推理系列产品