Searched refs:shape_attrs (Results 1 – 1 of 1) sorted by relevance
590 std::vector<TensorShapeProto> shape_attrs; in ValidateShape() local592 if (!GetNodeAttr(node->attrs(), kAttrName, &shape_attrs).ok()) { in ValidateShape()599 if (shape_attrs.size() < node->num_outputs()) { in ValidateShape()603 shape_attrs.size(), " outputs"); in ValidateShape()608 if (shape_attrs.size() > node->num_outputs()) { in ValidateShape()611 << " attribute specifies shapes for " << shape_attrs.size() in ValidateShape()615 const TensorShapeProto& p = shape_attrs[i]; in ValidateShape()