Searched refs:input_shapes_attr (Results 1 – 2 of 2) sorted by relevance
166 std::map<string, std::tuple<int64, int64, int64>> input_shapes_attr; in FormatInputShapes() local186 auto t = input_shapes_attr.find(shape_type_str); in FormatInputShapes()187 if (t == input_shapes_attr.end()) { in FormatInputShapes()188 input_shapes_attr.insert( in FormatInputShapes()190 t = input_shapes_attr.find(shape_type_str); in FormatInputShapes()192 input_shapes_attr[shape_type_str] = std::make_tuple( in FormatInputShapes()196 if (input_shapes_attr.empty()) { in FormatInputShapes()201 shape_count_vec(input_shapes_attr.begin(), input_shapes_attr.end()); in FormatInputShapes()
60 input_shapes_attr = fdef.attr.get("_input_shapes", None)61 if input_shapes_attr is not None:62 input_shapes = input_shapes_attr.list.shape