Searched refs:mGWS_M (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/buffer/ |
| H A D | ConvBufWinograd.cpp | 202 mGWS_M.resize(total_num); in onResize()
277 mGWS_M[b] = {globalWorkSize[min_index][0], globalWorkSize[min_index][1]}; in onResize()
281 mMatMul[b].setArg(index++, mGWS_M[b][0]); in onResize()
282 mMatMul[b].setArg(index++, mGWS_M[b][1]); in onResize()
346 runKernel2D(mMatMul[index], mGWS_M[index], mLWS_M[index], in onExecute()
353 runKernel2D(mMatMul[index], mGWS_M[index], mLWS_M[index], in onExecute()
|
| H A D | ConvBufWinograd.hpp | 57 std::vector<std::vector<uint32_t> > mGWS_M; member in MNN::OpenCL::ConvBufWinograd
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/image/ |
| H A D | ConvWinograd.cpp | 257 mGWS_M.resize(total_num); in onResize()
332 …mGWS_M[index] = {static_cast<uint32_t>(gemmWidth*gemmHeight), static_cast<uint32_t>(alpha * alpha)… in onResize()
334 …mLWS_M[index] = localWS2DDefault(mGWS_M[index], mMaxWGS_M[index], mOpenCLBackend->getOpenCLRuntime… in onResize()
383 runKernel2D(mMatMul[index], mGWS_M[index], mLWS_M[index], in onExecute()
390 runKernel2D(mMatMul[index], mGWS_M[index], mLWS_M[index], in onExecute()
|
| H A D | ConvWinograd.hpp | 56 std::vector<std::vector<uint32_t> > mGWS_M; member in MNN::OpenCL::ConvWinograd
|