megengine.jit.GraphOptimizationConfig

class GraphOptimizationConfig[源代码]

Configuration for graph optimization: False for OFF, True for ON. The default value None means that opt_level will decide whther this optimization will be applied or not.

参数
  • jit_fuse_dimshuffle – whether to fuse dimshuffle in JIT optimization

  • jit_fuse_reduce – whether to fuse reduce in JIT optimization

Methods