Home
last modified time | relevance | path

Searched defs:batch (Results 1 – 9 of 9) sorted by relevance

/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgInterfaces.h44 SmallVector<unsigned, 2> batch; member
76 SmallVector<unsigned, 2> batch; member
/llvm-project/mlir/unittests/Dialect/Utils/
H A DStructuredOpsUtilsTest.cpp194 AffineExpr batch, m, n, k; in TEST() local
207 AffineExpr batch, m, n, k; in TEST() local
220 AffineExpr batch, m, n, k; in TEST() local
233 AffineExpr batch, m, n, k; in TEST() local
324 AffineExpr batch, k, n; in TEST() local
337 AffineExpr batch, k, n; in TEST() local
363 AffineExpr batch, k, n; in TEST() local
376 AffineExpr batch, k, n; in TEST() local
389 AffineExpr batch, k, n; in TEST() local
415 AffineExpr batch, k, n; in TEST() local
/llvm-project/compiler-rt/lib/tsan/rtl/
H A Dtsan_dense_alloc.h122 IndexT batch; member
180 T *batch = (T *)MmapOrDie(kL2Size * sizeof(T), name_); in AllocSuperBlock() local
/llvm-project/compiler-rt/lib/sanitizer_common/tests/
H A Dsanitizer_quarantine_test.cpp38 while (QuarantineBatch *batch = cache->DequeueBatch()) in DeallocateCache() local
/llvm-project/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeTransposeConv.cpp175 int64_t batch = inputTy.getDimSize(0); matchAndRewrite() local
/llvm-project/compiler-rt/lib/sanitizer_common/
H A Dsanitizer_quarantine.h32 void *batch[kSize]; member
H A Dsanitizer_allocator_local_cache.h211 void *batch[2 * TransferBatch::kMaxNumCached]; member
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp641 llvm::SmallDenseSet<int64_t> batch = inputExprWalker.unConvolvedDims; inferConvolutionDimsImpl() local
[all...]
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp1439 auto batch = inputTy.getDimSize(0); matchAndRewrite() local
1561 Value batch = b.create<linalg::IndexOp>(0); matchAndRewrite() local