Searched refs:kFirstDimSize (Results 1 – 2 of 2) sorted by relevance
81 constexpr int64_t kFirstDimSize = 2; variable
728 …if (attrs.count(kGradIndex) == 0 || attrs.count(kIndiceIndex) == 0 || attrs.count(kFirstDimSize) =… in SparsePartitioner()736 iter = attrs.find(kFirstDimSize); in SparsePartitioner()