:py:mod:`neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer`
=====================================================================================================

.. py:module:: neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer

.. autoapi-nested-parse::

   ExpandDims Graph Rewriter.



Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer.ExpandDimsOptimizer




.. py:class:: ExpandDimsOptimizer(model)




   Calculate ExpandDims and remove it if its input is weight and next node is Conv2D.