`neural_compressor.ux.components.model.tensorflow.utils`

TensorFlow model utils.

Module Contents

Functions

get_input_shape(→ List[int])

Get input shape of passed graph.

neural_compressor.ux.components.model.tensorflow.utils.get_input_shape(graph_def: Any, fix_dynamic_shape: int) → List[int][source]: Get input shape of passed graph.