GraphOptimizationConfig¶
- class GraphOptimizationConfig[source]¶
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.
- Parameters
jit_fuse_dimshuffle – whether to fuse dimshuffle in JIT optimization
jit_fuse_reduce – whether to fuse reduce in JIT optimization