Searched refs:CheckLeafCount (Results 1 – 2 of 2) sorted by relevance
208 Status HloSharding::CheckLeafCount(const Shape& shape) const { in CheckLeafCount() function in xla::HloSharding220 TF_RETURN_IF_ERROR(CheckLeafCount(shape)); in AsShapeTree()238 TF_RETURN_IF_ERROR(CheckLeafCount(shape)); in GetTupleSharding()276 TF_RETURN_IF_ERROR(CheckLeafCount(shape)); in ValidateTuple()
249 Status CheckLeafCount(const Shape& shape) const;