neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_leakyrelu
Convert LeakyRelu Graph Rewriter.
Module Contents
Classes
Convert below subgraph to Node A + LeakyRelu. |
- class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_leakyrelu.ConvertLeakyReluOptimizer(model)[source]
Convert below subgraph to Node A + LeakyRelu.
- Node A Node A
- x |x |Mul —> |x |x |
Maximum LeakyRelu
Note, the coefficient of Mul should be less than 1 or the conversion is not valid.