Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/estimator/
D__init__.py36 keras_model=None, argument
173 keras_model=keras_model,
185 def model_to_estimator_v2(keras_model=None, argument
361 keras_model=keras_model,
/external/tensorflow/tensorflow/python/keras/applications/
Defficientnet_weight_update_util.py51 def write_ckpt_to_h5(path_h5, path_ckpt, keras_model, use_ema=True): argument
64 model_name_keras = keras_model.name
67 keras_weight_names = [w.name for w in keras_model.weights]
82 for w in keras_model.weights:
116 total_weights = len(keras_model.weights)
118 keras_model.save_weights(path_h5)
373 write_ckpt_to_h5(args.output, args.ckpt, keras_model=model)
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_quantization_test.py154 keras_model = keras.models.load_model(
156 keras_model.inputs[0].set_shape([1, 299, 299, 3])
158 converter = _lite.TFLiteConverterV2.from_keras_model(keras_model)
Dmodel_coverage_lib.py400 keras_model = keras.models.load_model(filename)
401 return lambda input_data: [keras_model.predict(input_data)]
512 keras_model,
542 tf_results = keras_model.predict(input_data)
544 tf_results = keras_model.predict(input_data[0])
971 keras_model = keras.models.load_model(filename)
973 for tensor, shape in zip(keras_model.inputs, input_shapes):
976 converter = _lite.TFLiteConverterV2.from_keras_model(keras_model)
/external/tensorflow/tensorflow/python/keras/benchmarks/
Dbenchmark_util.py67 def get_keras_examples_metadata(keras_model, argument
72 'implementation': keras_model + impl,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.estimator.pbtxt5 …argspec: "args=[\'keras_model\', \'keras_model_path\', \'custom_objects\', \'model_dir\', \'config…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.estimator.pbtxt5 …argspec: "args=[\'keras_model\', \'keras_model_path\', \'custom_objects\', \'model_dir\', \'config…
/external/tensorflow/tensorflow/lite/python/
Dlite.py807 def __init__(self, keras_model, trackable_obj=None): argument
819 self._keras_model = keras_model
1534 keras_model = keras_deps.get_load_model_function()(
1536 function = _trace_model_call(keras_model)
1542 self._keras_model = keras_model
1551 keras_model = keras_deps.get_load_model_function()(
1559 input_tensors = keras_model.inputs
1564 output_tensors = keras_model.outputs
1568 self._keras_model = keras_model
Dlite_test.py2241 keras_model = keras.models.load_model(self._keras_file)
2242 keras_result = keras_model.predict(input_data)
2271 keras_model = keras.models.load_model(
2273 keras_result = keras_model.predict(input_data)
2393 keras_model = keras.models.load_model(self._keras_file)
2394 keras_result = keras_model.predict(input_data)
2598 keras_model = keras.models.load_model(self._keras_file)
2599 keras_result = keras_model.predict(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/python/keras/
Dcallbacks_test.py2699 def keras_model(self, *args, **kwargs): member in SummaryOpsTest
2715 event = self.keras_model(name='my_name', data=model, step=1)
2726 event = self.keras_model(name='my_name', data=model)