Home
last modified time | relevance | path

Searched refs:filterDataPtr (Results 1 – 16 of 16) sorted by relevance

/dports/misc/mnn/MNN-1.2.0/source/backend/hiai/execution/
H A DNPUConvolution.cpp35 const float *filterDataPtr = nullptr; in onResize() local
47 filterDataPtr = quanCommon->weightFloat.get(); in onResize()
51 if (nullptr == filterDataPtr) { in onResize()
53 filterDataPtr = conv2D->weight()->data(); in onResize()
69 filter->SetData((uint8_t *)filterDataPtr, weightSize * sizeof(float)); in onResize()
H A DNPUConvolutionDepthwise.cpp35 const float *filterDataPtr = nullptr; in onResize() local
47 filterDataPtr = quanCommon->weightFloat.get(); in onResize()
51 if (nullptr == filterDataPtr) { in onResize()
53 filterDataPtr = conv2D->weight()->data(); in onResize()
69 filter->SetData((uint8_t *)filterDataPtr, weightSize * sizeof(float)); in onResize()
H A DNPUDeconvolution.cpp35 const float *filterDataPtr = nullptr; in onResize() local
37 if (nullptr == filterDataPtr) { in onResize()
39 filterDataPtr = conv2D->weight()->data(); in onResize()
55 filter->SetData((uint8_t *)filterDataPtr, weightSize * sizeof(float)); in onResize()
/dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/image/
H A DConvExecution.cpp82 const float *filterDataPtr = nullptr; in ConvExecution() local
94 filterDataPtr = quanCommon->weightFloat.get(); in ConvExecution()
100 if (nullptr == filterDataPtr) { in ConvExecution()
102 filterDataPtr = conv2dParams->weight()->data(); in ConvExecution()
153 … ((half_float::half*)kernelBufferPtr)[bufferIdx] = (half_float::half)(filterDataPtr[filterIdx]); in ConvExecution()
155 ((float*)kernelBufferPtr)[bufferIdx] = (float)(filterDataPtr[filterIdx]); in ConvExecution()
216 ((half_float::half*)ptrCL)[i] = (half_float::half)(filterDataPtr[i]); in ConvExecution()
219 ::memcpy(ptrCL, filterDataPtr, filterBuffer->size()); in ConvExecution()
H A DDepthwiseConvExecution.cpp45 const float* filterDataPtr = nullptr; in DepthwiseConvExecution() local
48 … ConvolutionCommon::getConvParameters(&quanCommon, mCon2dParams, &filterDataPtr, &filterDataSize); in DepthwiseConvExecution()
66 ((half_float::half *)ptrCL)[i] = (half_float::half)(filterDataPtr[i]); in DepthwiseConvExecution()
69 ::memcpy(ptrCL, filterDataPtr, filterBuffer->size()); in DepthwiseConvExecution()
H A DDepthwiseDeconvExecution.cpp43 const float* filterDataPtr = nullptr; in DepthwiseDeconvExecution() local
46 … ConvolutionCommon::getConvParameters(&quanCommon, mCon2dParams, &filterDataPtr, &tempWeightSize); in DepthwiseDeconvExecution()
64 ((half_float::half*)ptrCL)[i] = (half_float::half)(filterDataPtr[i]); in DepthwiseDeconvExecution()
67 ::memcpy(ptrCL, filterDataPtr, filterBuffer->size()); in DepthwiseDeconvExecution()
H A DConvWinograd.cpp46 const float* filterDataPtr = nullptr; in ConvWinograd() local
58 filterDataPtr = quanCommon->weightFloat.get(); in ConvWinograd()
62 if (nullptr == filterDataPtr) { in ConvWinograd()
64 filterDataPtr = op->weight()->data(); in ConvWinograd()
112 … Tensor::create<float>(std::vector<int>{co, ci, ky, kx}, (void*)(filterDataPtr), Tensor::CAFFE)); in ConvWinograd()
H A DDeconvExecution.cpp42 const float* filterDataPtr = nullptr; in DeconvExecution() local
45 ConvolutionCommon::getConvParameters(&quanCommon, conv2dParams, &filterDataPtr, &weightSize); in DeconvExecution()
52 …IOHW2OIHW<float, int>(filterDataPtr, filterDataPtrTransformed.data(), outputChannel, inputChannel,… in DeconvExecution()
/dports/misc/mnn/MNN-1.2.0/source/backend/vulkan/execution/
H A DVulkanDeconvolution.cpp34 const float* filterDataPtr = nullptr; in VulkanDeconvolution() local
37 ConvolutionCommon::getConvParameters(&quanCommon, conv, &filterDataPtr, &tempWeightSize); in VulkanDeconvolution()
39 if (nullptr != filterDataPtr) { in VulkanDeconvolution()
41 …ffer(vkBn->getMemoryPool(), false, ci * kh * kw * co * sizeof(float), filterDataPtr, VK_BUFFER_USA… in VulkanDeconvolution()
/dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/buffer/
H A DDeconvBufExecution.cpp43 const float* filterDataPtr = nullptr; in DeconvBufExecution() local
46 ConvolutionCommon::getConvParameters(&quanCommon, conv2dParams, &filterDataPtr, &weightSize); in DeconvBufExecution()
53 …IOHW2OIHW<float, int>(filterDataPtr, filterDataPtrTransformed.data(), outputChannel, inputChannel,… in DeconvBufExecution()
H A DDepthwiseConvBufExecution.cpp34 const float* filterDataPtr = nullptr; in DepthwiseConvBufExecution() local
37 … ConvolutionCommon::getConvParameters(&quanCommon, mCon2dParams, &filterDataPtr, &filterDataSize); in DepthwiseConvBufExecution()
55 ((half_float::half *)ptrCL)[i] = (half_float::half)(filterDataPtr[i]); in DepthwiseConvBufExecution()
58 ::memcpy(ptrCL, filterDataPtr, filterBuffer->size()); in DepthwiseConvBufExecution()
H A DConvBufExecution.hpp46 void setConv1x1WeightBuffer(int packCout, int packCin, const float* filterDataPtr);
H A DConvBufWinograd.cpp45 const float* filterDataPtr = nullptr; in ConvBufWinograd() local
47 ConvolutionCommon::getConvParameters(&quanCommon, op, &filterDataPtr, &weightSize); in ConvBufWinograd()
90 … Tensor::create<float>(std::vector<int>{oc, ic, ky, kx}, (void*)(filterDataPtr), Tensor::CAFFE)); in ConvBufWinograd()
H A DConvBufExecution.cpp227 void ConvBufExecution::setConv1x1WeightBuffer(int packCout, int packCin, const float* filterDataPtr in setConv1x1WeightBuffer() argument
246 … ((half_float::half*)kernelBufferPtr)[bufferIdx] = (half_float::half)(filterDataPtr[filterIdx]); in setConv1x1WeightBuffer()
248 ((float*)kernelBufferPtr)[bufferIdx] = (float)(filterDataPtr[filterIdx]); in setConv1x1WeightBuffer()
/dports/misc/mnn/MNN-1.2.0/source/backend/cuda/execution/
H A DDeconvSingleInputExecution.cu80 const float* filterDataPtr = nullptr; in DeconvSingleInputExecution() local
83 ConvolutionCommon::getConvParameters(&quanCommon, conv, &filterDataPtr, &weightSize); in DeconvSingleInputExecution()
87 … cuda_check(cudaMemcpy(mFilter, filterDataPtr, weightSize*sizeof(float), cudaMemcpyHostToDevice)); in DeconvSingleInputExecution()
H A DConvSingleInputExecution.cu85 const float* filterDataPtr = nullptr; in ConvSingleInputExecution() local
88 ConvolutionCommon::getConvParameters(&quanCommon, conv, &filterDataPtr, &weightSize); in ConvSingleInputExecution()
92 … cuda_check(cudaMemcpy(mFilter, filterDataPtr, weightSize*sizeof(float), cudaMemcpyHostToDevice)); in ConvSingleInputExecution()