torch_npu.jit.optimize
torch_npu.jit.optimize(jit_mod)
实现jit_mod优化。
参数解释:
jit_mod:用于被优化的ScriptFunction or ScriptModule。
返回值:
无
约束条件:
无
示例:
无
父主题:
torch_npu.jit