neural_compressor.ux.components.model.tensorflow.keras

Tensorflow keras model.

Module Contents

Classes

KerasModel

Keras model.

class neural_compressor.ux.components.model.tensorflow.keras.KerasModel(path: str)

Bases: neural_compressor.ux.components.model.tensorflow.model.TensorflowModel

Keras model.

get_input_nodes() → List[Any] | None

Get model input nodes.

get_output_nodes() → List[Any] | None

Get model output nodes.

static supports_path(path: str) → bool

Check if given path is of supported model.