Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
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 = str_util::Join(input_layer_shapes, ":"); in PrintBenchmarkUsage()
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc440 std::vector<string> input_layer_shapes = in Main() local
448 if ((input_layers.size() != input_layer_shapes.size()) || in Main()
459 << input_layer_shapes.size() << " items)"; in Main()
536 CHECK(str_util::SplitAndParseAsInts(input_layer_shapes[n], ',', &sizes)) in Main()
537 << "Incorrect size string specified: " << input_layer_shapes[n]; in Main()