torch_npu.jit.optimize

API接口

torch_npu.jit.optimize(jit_mod)

功能描述

实现jit_mod优化。

参数说明

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