Home
last modified time | relevance | path

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

/dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/buffer/
H A DConvBufWinograd.cpp234 auto gemmHeight = ocC4; in onResize() local
254 …ze[knl_idx] = {static_cast<uint32_t>(UP_DIV(gemmWidth, itemW[knl_idx])*gemmHeight), static_cast<ui… in onResize()
262 kernel[knl_idx].setArg(index++, gemmHeight); in onResize()
287 mMatMul[b].setArg(index++, gemmHeight); in onResize()
/dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/image/
H A DConvWinograd.cpp331 auto gemmHeight = ocC4; in onResize() local
332 …mGWS_M[index] = {static_cast<uint32_t>(gemmWidth*gemmHeight), static_cast<uint32_t>(alpha * alpha)… in onResize()