Searched defs:AlignSizeForLaunchKernel (Results 1 – 6 of 6) sorted by relevance
26 size_t AscendLaunchKernel::AlignSizeForLaunchKernel(size_t size) { return MemoryManager::GetCommonA… in AlignSizeForLaunchKernel() function in mindspore::device::ascend::AscendLaunchKernel
26 size_t AscendLaunchMul::AlignSizeForLaunchKernel(size_t size) { in AlignSizeForLaunchKernel() function in mindspore::device::ascend::AscendLaunchMul
27 size_t AscendLaunchTransData::AlignSizeForLaunchKernel(size_t size) { in AlignSizeForLaunchKernel() function in mindspore::device::ascend::AscendLaunchTransData
26 size_t AscendLaunchAtomicClean::AlignSizeForLaunchKernel(size_t size) { in AlignSizeForLaunchKernel() function in mindspore::device::ascend::AscendLaunchAtomicClean
27 size_t GPULaunchMul::AlignSizeForLaunchKernel(size_t size) { return GPULaunchkernel::AlignSizeForLa… in AlignSizeForLaunchKernel() function in mindspore::device::gpu::GPULaunchMul
40 size_t GPULaunchkernel::AlignSizeForLaunchKernel(size_t size) { return AlignMemorySize(size); } in AlignSizeForLaunchKernel() function in mindspore::device::gpu::GPULaunchkernel