Lines Matching refs:mUnits

62     mUnits.resize(1);  in onResize()
73 …mUnits[0].kernel = runTime->buildKernel("binary_buf", "prelu_buf", {"-DOPERATOR=select(in0*in1,in0…  in onResize()
74     mMaxWorkGroupSize      = static_cast<uint32_t>(runTime->getMaxWorkGroupSize(mUnits[0].kernel));  in onResize()
78     mUnits[0].kernel.setArg(index++, globalSize[0]);  in onResize()
79     mUnits[0].kernel.setArg(index++, globalSize[1]);  in onResize()
80     mUnits[0].kernel.setArg(index++, openCLBuffer(inputs[0]));  in onResize()
81     mUnits[0].kernel.setArg(index++, openCLBuffer(mPreluParam.get()));  in onResize()
82     mUnits[0].kernel.setArg(index++, openCLBuffer(outputs[0]));  in onResize()
83     mUnits[0].kernel.setArg(index++, nhwcArray);  in onResize()
86 …t(globalSize, mMaxWorkGroupSize, mOpenCLBackend->getOpenCLRuntime(), name, mUnits[0].kernel).first;  in onResize()
88     mUnits[0].globalWorkSize = {globalSize[0], globalSize[1]};  in onResize()
89     mUnits[0].localWorkSize  = {localSize[0], localSize[1]};  in onResize()