Searched refs:wanted_shapes (Results 1 – 1 of 1) sorted by relevance
117 const std::set<string>& wanted_shapes, Session* session, in GetOutputShapes() argument123 for (const string& wanted_shape : wanted_shapes) { in GetOutputShapes()156 std::set<string> wanted_shapes; in CalculateFlops() local160 wanted_shapes.insert(input); in CalculateFlops()162 wanted_shapes.insert(node.name()); in CalculateFlops()167 GetOutputShapes(inputs, wanted_shapes, session, &found_shapes)); in CalculateFlops()