Searched refs:IsSparseArray (Results 1 – 11 of 11) sorted by relevance
32 if (LayoutUtil::IsSparseArray(subshape)) { in Run()
113 static bool IsSparseArray(const Shape& shape);
577 if (LayoutUtil::IsSparseArray(shape)) { in ByteSizeOf()604 if (LayoutUtil::IsSparseArray(shape)) { in ByteSizeOfElements()616 CHECK(LayoutUtil::IsSparseArray(shape)); in ByteSizeOfSparseIndices()661 if (LayoutUtil::IsSparseArray(shape) && shape.rank() == 0) { in ValidateShapeWithOptionalLayoutInternal()690 if (shape_has_valid_layout && LayoutUtil::IsSparseArray(shape)) { in ValidateShapeSize()
370 return LayoutUtil::IsSparseArray(subshape()) in element_count()953 LayoutUtil::IsSparseArray(ShapeUtil::GetSubshape(shape(), shape_index))); in GetSparseElement()963 CHECK(LayoutUtil::IsSparseArray(subshape)); in AppendSparseElement()1059 CHECK(LayoutUtil::IsSparseArray(shape())); in PopulateSparse()
124 if (LayoutUtil::IsSparseArray(shape)) { in SetPiece()836 CHECK(LayoutUtil::IsSparseArray(subshape)); in GetSparseElementAsString()1032 CHECK(LayoutUtil::IsSparseArray(subshape())); in SortSparseElementsInternal()1179 } else if (LayoutUtil::IsSparseArray(subshape)) { in ToStringHelper()1192 CHECK(LayoutUtil::IsSparseArray(shape())); in sparse_element_count()1973 if (LayoutUtil::IsSparseArray(subshape())) { in CopyFromProto()2095 if (LayoutUtil::IsSparseArray(shape())) { in ToProto()
293 /* static */ bool LayoutUtil::IsSparseArray(const Shape& shape) { in IsSparseArray() function in xla::LayoutUtil
89 EXPECT_FALSE(Match(&array_shape, match::Shape().IsSparseArray())); in TEST()112 EXPECT_TRUE(Match(&array_shape, match::Shape().IsSparseArray())); in TEST()
34 if (LayoutUtil::IsSparseArray(subshape)) { in VerifyNotSparse()
1062 constexpr auto IsSparseArray() const
2161 if (LayoutUtil::IsSparseArray(shape)) { in ParseNonTupleLiteral()