Searched refs:wanted_shapes (Results 1 – 1 of 1) sorted by relevance
118 const std::set<string>& wanted_shapes, Session* session, in GetOutputShapes() argument124 for (const string& wanted_shape : wanted_shapes) { in GetOutputShapes()157 std::set<string> wanted_shapes; in CalculateFlops() local161 wanted_shapes.insert(input); in CalculateFlops()163 wanted_shapes.insert(node.name()); in CalculateFlops()168 GetOutputShapes(inputs, wanted_shapes, session, &found_shapes)); in CalculateFlops()