Searched defs:tensorRank (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | CudaRuntimeWrappers.cpp | 395 mgpuTensorMapEncodeTiled(CUtensorMap * tensorMap,CUtensorMapDataType tensorDataType,cuuint32_t tensorRank,void * globalAddress,const cuuint64_t * globalDim,const cuuint64_t * globalStrides,const cuuint32_t * boxDim,const cuuint32_t * elementStrides,CUtensorMapInterleave interleave,CUtensorMapSwizzle swizzle,CUtensorMapL2promotion l2Promotion,CUtensorMapFloatOOBfill oobFill) mgpuTensorMapEncodeTiled() argument
450 mgpuTensorMapEncodeTiledMemref(int64_t tensorRank,void * rankedDescriptor,const CUtensorMapDataType tensorDataType,CUtensorMapInterleave interleave,CUtensorMapSwizzle swizzle,CUtensorMapL2promotion l2Promotion,CUtensorMapFloatOOBfill oobFill,int64_t * inputBoxDims) mgpuTensorMapEncodeTiledMemref() argument
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 2522 unsigned tensorRank = insertOp.getType().getRank(); rewriteUser() local
|