(
beta
)torch_npu.jit.optimize
接口原型
torch_npu.jit.optimize(jit_mod)
功能描述
实现jit_mod优化。
参数说明
jit_mod:用于被优化的ScriptFunction or ScriptModule。
父主题:
torch_npu.jit