Home
last modified time | relevance | path

Searched refs:threadgroupsPerGrid (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/
H A DMPSImageUtils.mm59 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
82 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
144 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
164 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
196 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
228 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMPSCNNUtils.mm26 const auto threadgroupsPerGrid = MTLSizeMake(
32 return {threadsPerThreadgroup, threadgroupsPerGrid, threadsPerGrid};
H A DMPSCNNUtils.h27 MTLSize threadgroupsPerGrid; member
H A DMPSCNNClampOp.mm40 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/ops/
H A DMetalConcat.mm45 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
121 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
154 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMetalLeakyReLU.mm41 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
76 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMetalHardshrink.mm43 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
78 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMetalHardswish.mm39 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
72 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMetalCopy.mm44 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMetalBinaryElementwise.mm92 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
130 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMetalPadding.mm79 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMetalChunk.mm52 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMetalTranspose.mm80 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
H A DMetalReshape.mm56 [encoder dispatchThreadgroups:launchParams.threadgroupsPerGrid
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DBucketization.mm244 …MTLSize threadgroupsPerGrid = MTLSizeMake(std::min(maxThreadgroups, ceil_div<int64_t>(numel_in, tg…
246 … [computeEncoder dispatchThreadgroups:threadgroupsPerGrid threadsPerThreadgroup:threadGroupSize];