Home
last modified time | relevance | path

Searched refs:c_32_block (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/
H A Doffsets.h66 int c_32_block = OC % 32 ? (32 + OC - (OC % 32)) : OC; in off_nCdhw32c() local
68 off += n * (c_32_block / 32) * G * D * H * W * 32; in off_nCdhw32c()
92 int c_32_block = OC % 32 ? (32 + OC - (OC % 32)) : OC; in off_NCdhw32n32c() local
94 off += (n / 32) * (c_32_block / 32) * D * H * W * 32 * 32; in off_NCdhw32n32c()
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/
H A Doffsets.h66 int c_32_block = OC % 32 ? (32 + OC - (OC % 32)) : OC; in off_nCdhw32c() local
68 off += n * (c_32_block / 32) * G * D * H * W * 32; in off_nCdhw32c()
92 int c_32_block = OC % 32 ? (32 + OC - (OC % 32)) : OC; in off_NCdhw32n32c() local
94 off += (n / 32) * (c_32_block / 32) * D * H * W * 32 * 32; in off_NCdhw32n32c()