Searched refs:input_layers (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 440 std::vector<string> input_layers = str_util::Split(input_layer_string, ','); in Main() local
449 if ((input_layers.size() != input_layer_shapes.size()) || in Main()
450 (input_layers.size() != input_layer_types.size())) { in Main()
456 << input_layers.size() << " items)"; in Main()
463 const size_t inputs_count = input_layers.size(); in Main()
551 input.name = input_layers[n]; in Main()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | summarize_graph_main.cc | 70 std::vector<string> input_layers; in PrintBenchmarkUsage() local
74 input_layers.push_back(node->name()); in PrintBenchmarkUsage()
106 string input_layer_value = absl::StrJoin(input_layers, ","); in PrintBenchmarkUsage()
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | sequence_feature_column_test.py | 624 input_layers = {
632 fc_layer, _ = ksfc.SequenceFeatures(cols)(input_layers)
639 model = keras.models.Model(input_layers, output)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 221 def _clone_layers_and_model_config(model, input_layers, layer_fn): argument
241 if layer in input_layers:
242 created_layers[layer.name] = input_layers[layer]
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | save_test.py | 165 input_layers = {
170 fc_layer = dense_features.DenseFeatures(cols)(input_layers)
173 model = keras.models.Model(input_layers, output)
201 input_layers = {
209 fc_layer, _ = ksfc.SequenceFeatures(cols)(input_layers)
216 model = keras.models.Model(input_layers, output)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | functional.py | 1301 input_layers = tf_utils.convert_inner_node_data(
1303 for layer_data in nest.flatten(input_layers):
1321 input_tensors = nest.pack_sequence_as(input_layers, input_tensors)
|