Home
last modified time | relevance | path

Searched defs:numTensors (Results 1 – 6 of 6) sorted by relevance

/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DCodegenEnv.cpp45 unsigned numTensors, unsigned numLoops, unsigned maxRank) in CodegenEnv()
H A DLoopEmitter.cpp135 const unsigned numTensors = numManifestTensors + 1; in initialize() local
236 for (TensorId t = 0, numTensors in initializeLoopEmit() local
301 for (TensorId t = 0, numTensors = getNumManifestTensors(); t < numTensors; initializeLoopEmit() local
[all...]
/llvm-project/mlir/unittests/Dialect/SparseTensor/
H A DMergerTest.cpp126 MergerTestBase(unsigned numTensors, unsigned numLoops) in MergerTestBase()
/llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/
H A DMerger.h628 const unsigned numTensors; variable
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsification.cpp1352 const unsigned numTensors = op->getNumOperands(); matchAndRewrite() local
H A DSparseGPUCodegen.cpp1261 const unsigned numTensors = op->getNumOperands(); in matchAndRewrite() local