Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc117 const std::set<string>& wanted_shapes, Session* session, in GetOutputShapes() argument
123 for (const string& wanted_shape : wanted_shapes) { in GetOutputShapes()
156 std::set<string> wanted_shapes; in CalculateFlops() local
160 wanted_shapes.insert(input); in CalculateFlops()
162 wanted_shapes.insert(node.name()); in CalculateFlops()
167 GetOutputShapes(inputs, wanted_shapes, session, &found_shapes)); in CalculateFlops()