GraphOptimizationConfig

class GraphOptimizationConfig

图优化的配置：False 关闭，True 开启。 默认值 None 意味着 opt_level 将决定是否应用此优化。

参数

• jit_fuse_dimshuffle – 在JIT优化中是否融合dimshuffle

• jit_fuse_reduce – 在JIT优化中是否融合reduce

上一页

DTRConfig

下一页

megengine.amp