Lines Matching refs:rank
40 os << "base@ = " << reinterpret_cast<void *>(V.data) << " rank = " << V.rank in printMemRefMetaData()
43 if (V.rank == 0) in printMemRefMetaData()
46 for (int64_t i = 1; i < V.rank; ++i) in printMemRefMetaData()
129 static void print(std::ostream &os, T *base, int64_t dim, int64_t rank,
132 static void printFirst(std::ostream &os, T *base, int64_t dim, int64_t rank,
135 static void printLast(std::ostream &os, T *base, int64_t dim, int64_t rank,
142 int64_t rank, int64_t offset,
146 print(os, base, dim - 1, rank, offset, sizes + 1, strides + 1);
159 int64_t rank, int64_t offset,
165 printFirst(os, base, dim, rank, offset, sizes, strides);
167 printSpace(os, rank - dim + 1);
168 print(os, base, dim - 1, rank, offset + i * strides[0], sizes + 1,
176 printLast(os, base, dim, rank, offset, sizes, strides);
181 int64_t rank, int64_t offset,
184 printSpace(os, rank - dim + 1);
185 print(os, base, dim - 1, rank, offset + (sizes[0] - 1) * (*strides),
194 if (M.rank == 0)
196 MemRefDataPrinter<T>::print(std::cout, M.data, M.rank, M.rank, M.offset,
198 if (M.rank == 0)
226 extern "C" MLIR_RUNNERUTILS_EXPORT void print_memref_i32(int64_t rank,
228 extern "C" MLIR_RUNNERUTILS_EXPORT void print_memref_i64(int64_t rank,
230 extern "C" MLIR_RUNNERUTILS_EXPORT void print_memref_f32(int64_t rank,
232 extern "C" MLIR_RUNNERUTILS_EXPORT void print_memref_f64(int64_t rank,