Home
last modified time | relevance | path

Searched refs:output_layers (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dsummarize_graph_main.cc101 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 = absl::StrJoin(output_layers, ","); in PrintBenchmarkUsage()
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc447 std::vector<string> output_layers = str_util::Split(output_layer_string, ','); in Main() local
575 inputs, output_layers, target_layers, session.get(), in Main()
590 inputs, output_layers, target_layers, session.get(), nullptr, in Main()
605 inputs, output_layers, target_layers, session.get(), stats.get(), in Main()
/external/libgav1/libgav1/src/
Ddecoder_impl.h123 } output_layers[kMaxLayers]; member
Ddecoder_impl.cc812 temporal_unit.output_layers[temporal_unit.output_layer_count - 1].frame); in DequeueFrame()
813 temporal_unit.output_layers[temporal_unit.output_layer_count - 1].frame = in DequeueFrame()
930 temporal_unit.output_layers, in ParseAndSchedule()
931 temporal_unit.output_layers + temporal_unit.output_layer_count); in ParseAndSchedule()
1007 temporal_unit.output_layers[temporal_unit.output_layer_count].frame = in DecodeFrame()
1009 temporal_unit.output_layers[temporal_unit.output_layer_count] in DecodeFrame()
/external/tensorflow/tensorflow/python/keras/engine/
Dfunctional.py1311 output_layers = tf_utils.convert_inner_node_data(
1313 for layer_data in nest.flatten(output_layers):
1322 output_tensors = nest.pack_sequence_as(output_layers, output_tensors)