Home
last modified time | relevance | path

Searched refs:keras_model (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/keras/estimator/
D__init__.py27 keras_model=None, argument
163 keras_model=keras_model,
175 def model_to_estimator_v2(keras_model=None, argument
350 keras_model=keras_model,
/external/tensorflow/tensorflow/python/keras/applications/
Defficientnet_weight_update_util.py46 def write_ckpt_to_h5(path_h5, path_ckpt, keras_model, use_ema=True): argument
59 model_name_keras = keras_model.name
62 keras_weight_names = [w.name for w in keras_model.weights]
77 for w in keras_model.weights:
111 total_weights = len(keras_model.weights)
113 keras_model.save_weights(path_h5)
368 write_ckpt_to_h5(args.output, args.ckpt, keras_model=model)
/external/tensorflow/tensorflow/python/keras/benchmarks/
Dbenchmark_util.py64 def get_keras_examples_metadata(keras_model, argument
69 'implementation': keras_model + impl,
/external/tensorflow/tensorflow/lite/python/
Dlite.py1025 def __init__(self, keras_model, trackable_obj=None): argument
1037 self._keras_model = keras_model
1921 keras_model = keras_deps.get_load_model_function()(model_file,
1923 function = _trace_model_call(keras_model)
1929 self._keras_model = keras_model
1938 keras_model = keras_deps.get_load_model_function()(model_file,
1946 input_tensors = keras_model.inputs
1951 output_tensors = keras_model.outputs
1955 self._keras_model = keras_model
Dlite_test.py2370 keras_model = keras.models.load_model(self._keras_file)
2371 keras_result = keras_model.predict(input_data)
2400 keras_model = keras.models.load_model(
2402 keras_result = keras_model.predict(input_data)
2522 keras_model = keras.models.load_model(self._keras_file)
2523 keras_result = keras_model.predict(input_data)
2727 keras_model = keras.models.load_model(self._keras_file)
2728 keras_result = keras_model.predict(input_data)
Dlite_v2_test.py1977 keras_model = tf.keras.Sequential([
1988 converter = lite.TFLiteConverterV2.from_keras_model(keras_model)
1994 expected_value = keras_model(input_data)
/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
Dpython_api.md55 converter = tf.compat.v1.lite.TFLiteConverter.from_keras_model_file('keras_model.h5')
86 keras_file = 'keras_model.h5'
Dcmdline_examples.md68 --keras_model_file=/tmp/keras_model.h5 \
/external/tensorflow/tensorflow/lite/g3doc/convert/
Drnn.md52 keras_model = build_keras_lstm(...)
55 converter = TFLiteConverter.from_keras_model(keras_model)
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dsignatures.ipynb275 "keras_model = tf.keras.Sequential(\n",
284 "converter = tf.lite.TFLiteConverter.from_keras_model(keras_model)\n",
/external/tensorflow/tensorflow/python/keras/
Dcallbacks_test.py2899 def keras_model(self, *args, **kwargs): member in SummaryOpsTest
2915 event = self.keras_model(name='my_name', data=model, step=1)
2926 event = self.keras_model(name='my_name', data=model)