Searched refs:AlignSizeForLaunchKernel (Results 1 – 16 of 16) sorted by relevance
27 size_t GPULaunchMul::AlignSizeForLaunchKernel(size_t size) { return GPULaunchkernel::AlignSizeForLa… in AlignSizeForLaunchKernel() function in mindspore::device::gpu::GPULaunchMul
31 size_t AlignSizeForLaunchKernel(size_t size) override;
33 size_t AlignSizeForLaunchKernel(size_t size) override;
40 size_t GPULaunchkernel::AlignSizeForLaunchKernel(size_t size) { return AlignMemorySize(size); } in AlignSizeForLaunchKernel() function in mindspore::device::gpu::GPULaunchkernel
26 size_t AscendLaunchMul::AlignSizeForLaunchKernel(size_t size) { in AlignSizeForLaunchKernel() function in mindspore::device::ascend::AscendLaunchMul27 return AscendLaunchKernel::AlignSizeForLaunchKernel(size); in AlignSizeForLaunchKernel()
26 size_t AscendLaunchAtomicClean::AlignSizeForLaunchKernel(size_t size) { in AlignSizeForLaunchKernel() function in mindspore::device::ascend::AscendLaunchAtomicClean27 return AscendLaunchKernel::AlignSizeForLaunchKernel(size); in AlignSizeForLaunchKernel()
27 size_t AscendLaunchTransData::AlignSizeForLaunchKernel(size_t size) { in AlignSizeForLaunchKernel() function in mindspore::device::ascend::AscendLaunchTransData28 return AscendLaunchKernel::AlignSizeForLaunchKernel(size); in AlignSizeForLaunchKernel()
34 size_t AlignSizeForLaunchKernel(size_t size) override;
26 size_t AscendLaunchKernel::AlignSizeForLaunchKernel(size_t size) { return MemoryManager::GetCommonA… in AlignSizeForLaunchKernel() function in mindspore::device::ascend::AscendLaunchKernel
37 size_t AlignSizeForLaunchKernel(size_t size) override;
43 size_t AlignSizeForLaunchKernel(size_t size) override;
28 auto size = AlignSizeForLaunchKernel(list[i]); in ObtainKernelAddress()48 auto input_size = AlignSizeForLaunchKernel(inputs_list[i]); in ObtainKernelInputs()
37 virtual size_t AlignSizeForLaunchKernel(size_t size) = 0;
35 virtual size_t AlignSizeForLaunchKernel(size_t size) = 0;
69 auto input_size = AlignSizeForLaunchKernel(size * 1); in ObtainMulInputsAddr()