neural_compressor.ux.components.model.tensorflow.utils¶
TensorFlow model utils.
Module Contents¶
Functions¶
|
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]¶
Get input shape of passed graph.