Searched defs:tensorDataType (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | CudaRuntimeWrappers.cpp | 452 descriptor->strides[Rank - i - 2] * elementSizeInBytes[tensorDataType]); in mgpuTensorMapEncodeTiledMemref() argument
394 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
434 mgpuGetMemRefDataAndShape(void * rawDescriptor,char ** addr,uint64_t * globalDim,uint64_t * globalStrides,const CUtensorMapDataType tensorDataType) mgpuGetMemRefDataAndShape() argument
[all...] |