Lines Matching defs:globalDim

404     const cuuint64_t *globalDim,        // Tensor size (number of elements)
415       tensorMap, tensorDataType, tensorRank, globalAddress, globalDim,
421               "globalDim[5]: %zu, %zu, %zu, %zu, %zu\n"
429               (void *)&tensorMap, tensorDataType, tensorRank, globalDim[0],
430               globalDim[1], globalDim[2], globalDim[3], globalDim[4],
440                                uint64_t *globalDim, uint64_t *globalStrides,
446     globalDim[i] = static_cast<uint64_t>(descriptor->sizes[Rank - i - 1]);
469   uint64_t globalDim[5] = {1, 1, 1, 1, 1}, globalStrides[5] = {0};
475     mgpuGetMemRefDataAndShape<1>(rankedDescriptor, &globalAddress, globalDim,
479     mgpuGetMemRefDataAndShape<2>(rankedDescriptor, &globalAddress, globalDim,
483     mgpuGetMemRefDataAndShape<3>(rankedDescriptor, &globalAddress, globalDim,
487     mgpuGetMemRefDataAndShape<4>(rankedDescriptor, &globalAddress, globalDim,
491     mgpuGetMemRefDataAndShape<5>(rankedDescriptor, &globalAddress, globalDim,
507                            globalAddress, globalDim, globalStrides, boxDim,