Searched refs:shapeAlloc (Results 1 – 2 of 2) sorted by relevance
65 ArrayRef<int64_t> shapeAlloc) { in makeStridedMemRefDescriptor() argument67 assert(shapeAlloc.size() == N); in makeStridedMemRefDescriptor()73 auto strides = makeStrides<N>(shapeAlloc); in makeStridedMemRefDescriptor()86 ArrayRef<int64_t> shapeAlloc = {}) {88 assert(shapeAlloc.size() == N);146 ArrayRef<int64_t> shape, ArrayRef<int64_t> shapeAlloc = {},155 if (shapeAlloc.empty()) in freeFunc()156 shapeAlloc = shape; in freeFunc()158 assert(shapeAlloc.size() == Rank); in freeFunc()160 assert(shape[i] <= shapeAlloc[i] && in freeFunc()[all …]
195 int64_t shapeAlloc[] = {k + 1, m + 1}; in TEST() local196 OwningMemRef<float, 2> a(shape, shapeAlloc, init); in TEST()252 int64_t shapeAlloc[] = {k + 1, m + 1}; in TEST() local253 OwningMemRef<float, 2> a(shape, shapeAlloc); in TEST()