Home
last modified time | relevance | path

Searched refs:input_layer_shapes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/
H A Dbenchmark_model.cc445 std::vector<string> input_layer_shapes = in Main() local
453 if ((input_layers.size() != input_layer_shapes.size()) || in Main()
464 << input_layer_shapes.size() << " items)"; in Main()
542 str_util::Split(input_layer_shapes[n], ','); in Main()
546 << "Incorrect size string specified: " << input_layer_shapes[n]; in Main()
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsummarize_graph_main.cc72 std::vector<string> input_layer_shapes; in PrintBenchmarkUsage() local
99 input_layer_shapes.push_back(sizes_string); in PrintBenchmarkUsage()
108 string input_layer_shape_value = absl::StrJoin(input_layer_shapes, ":"); in PrintBenchmarkUsage()