Searched defs:wanted_shapes (Results 1 – 1 of 1) sorted by relevance
117 const std::set<string>& wanted_shapes, Session* session, in GetOutputShapes()156 std::set<string> wanted_shapes; in CalculateFlops() local