Searched refs:layer_fn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 77 def _make_new_nodes(nodes_by_depth, layer_fn, layer_map, tensor_map): argument
101 new_layer = layer_fn(layer)
133 def _clone_functional_model(model, input_tensors=None, layer_fn=_clone_layer): argument
193 if not callable(layer_fn):
197 model, new_input_layers, layer_fn)
221 def _clone_layers_and_model_config(model, input_layers, layer_fn): argument
246 created_layers[layer.name] = layer_fn(layer)
283 def _clone_sequential_model(model, input_tensors=None, layer_fn=_clone_layer): argument
316 if not callable(layer_fn):
331 if isinstance(layer, InputLayer) else layer_fn(layer))
[all …]
|
| D | models_test.py | 106 keras.models._clone_sequential_model, layer_fn=models.share_weights)
162 keras.models._clone_functional_model, layer_fn=models.share_weights)
228 keras.models._clone_functional_model, layer_fn=models.share_weights)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils_v1.py | 727 model, input_tensors=inputs, layer_fn=models.share_weights)
730 model, input_tensors=inputs, layer_fn=models.share_weights)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_test.py | 1124 self.layer_fn = layers.Dense
|