neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_layout
Convert Layout Graph Rewriter.
Module Contents
Classes
The layout conversion optimizer, convert NCHW to NHWC format. |
- class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_layout.ConvertLayoutOptimizer(model, outputs)[source]
The layout conversion optimizer, convert NCHW to NHWC format.
It is executed only when NCHW node exists and tensorflow version is 2.4.0 and above.
- Args: model: input graph_def
outputs: output name list
Return: converted graph_def