Searched refs:shapeAlloc (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/include/mlir/ExecutionEngine/ |
| H A D | MemRefUtils.h | 66 ArrayRef<int64_t> shapeAlloc) { in makeStridedMemRefDescriptor() argument
68 assert(shapeAlloc.size() == N); in makeStridedMemRefDescriptor()
74 auto strides = makeStrides<N>(shapeAlloc); in makeStridedMemRefDescriptor()
87 ArrayRef<int64_t> shapeAlloc = {}) {
89 assert(shapeAlloc.size() == N);
147 ArrayRef<int64_t> shape, ArrayRef<int64_t> shapeAlloc = {},
156 if (shapeAlloc.empty()) in freeFunc()
157 shapeAlloc = shape; in freeFunc()
159 assert(shapeAlloc.size() == Rank); in freeFunc()
161 assert(shape[i] <= shapeAlloc[i] && in freeFunc()
[all …]
|
| /llvm-project/mlir/unittests/ExecutionEngine/ |
| H A D | Invoke.cpp | 207 int64_t shapeAlloc[] = {k + 1, m + 1}; in TEST()
208 OwningMemRef<float, 2> a(shape, shapeAlloc, init); in TEST()
265 int64_t shapeAlloc[] = {k + 1, m + 1}; in TEST()
266 OwningMemRef<float, 2> a(shape, shapeAlloc); in TEST()
198 int64_t shapeAlloc[] = {k + 1, m + 1}; TEST() local
256 int64_t shapeAlloc[] = {k + 1, m + 1}; TEST() local
|