Searched refs:output_layers (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | summarize_graph_main.cc | 101 std::vector<string> output_layers; in PrintBenchmarkUsage() local
102 output_layers.reserve(outputs.size()); in PrintBenchmarkUsage()
104 output_layers.push_back(node->name()); in PrintBenchmarkUsage()
109 string output_layer_value = str_util::Join(output_layers, ","); in PrintBenchmarkUsage()
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 446 std::vector<string> output_layers = str_util::Split(output_layer_string, ','); in Main() local
564 inputs, output_layers, target_layers, session.get(), in Main()
579 inputs, output_layers, target_layers, session.get(), nullptr, in Main()
594 inputs, output_layers, target_layers, session.get(), stats.get(), in Main()
|
| /external/tensorflow/tensorflow/examples/ios/benchmark/ |
| D | BenchmarkViewController.mm | 143 std::vector<tensorflow::Tensor>* output_layers,
154 output_layers, &run_metadata);
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | network.py | 1261 output_layers = tf_utils.convert_inner_node_data(
1263 for layer_data in nest.flatten(output_layers):
1271 output_tensors = nest.pack_sequence_as(output_layers, output_tensors)
|