Searched refs:totalWeightSize (Results 1 – 3 of 3) sorted by relevance
27 …auto totalWeightSize = ALIGN_UP4(mCommon->outputCount()) * ALIGN_UP4(mInputDepth) * (mCommon->kern… in GLConvolutionIm2col() local29 …o mKernelBuffer = std::shared_ptr<GLSSBOBuffer>(new GLSSBOBuffer(sizeof(float) * totalWeightSize)); in GLConvolutionIm2col()37 ::memset(dest, 0, totalWeightSize * sizeof(float)); in GLConvolutionIm2col()
58 auto totalWeightSize = in GLConvolution() local71 …o mKernelBuffer = std::shared_ptr<GLSSBOBuffer>(new GLSSBOBuffer(sizeof(float) * totalWeightSize)); in GLConvolution()
28 …auto totalWeightSize = ALIGN_UP4(mCommon->outputCount()) * (mCommon->kernelY() * mCommon->kernelX(… in _createBufferForConvDepthwise() local29 …:make_shared<VulkanBuffer>(extra->getMemoryPool(), false, sizeof(float) * totalWeightSize, nullptr, in _createBufferForConvDepthwise()