megengine.jit

>>> from megengine.jit import trace

trace	包装一个可调用对象，并提供以下功能：
exclude_from_trace
xla_trace	Wraps a callable, and provides accelerated evaluation compiled by xla.

Trace Config

SublinearMemoryConfig	亚线性内存优化的配置。
DTRConfig	DTR 内存优化的配置。
GraphOptimizationConfig	图优化的配置：False 关闭，True 开启。

上一页

megengine.optimizer.clip_grad_value

下一页

trace