Searched refs:keras_model (Results 1 – 1 of 1) sorted by relevance
71 keras_model = op_type.generate_keras_model(shapes, params)73 keras_model,151 def convert_keras_to_tflite(output_fpath, keras_model, quantize, dtype, bias_dtype, shape): argument153 keras_model.compile(loss=keras.losses.categorical_crossentropy,156 n_inputs = len(keras_model.inputs)157 converter = tf.lite.TFLiteConverter.from_keras_model(keras_model)
