Home
last modified time | relevance | path

Searched refs:weightSrc (Results 1 – 11 of 11) sorted by relevance

/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/compute/
H A DWinogradInt8Helper.hpp23 std::shared_ptr<Tensor> allocTransformWeight(const Tensor* weightSrc);
24 bool transformWeight(const Tensor* weightSrc, Tensor* weightDst);
30 …static bool weightOverflow(const Tensor* weightSrc, int unitY, int unitX, const Convolution2DCommo…
H A DConvInt8TiledExecutor.cpp42 auto weightSrc = weightOrigin->host<int8_t>(); in reorderWeight() local
46 const auto srcK = weightSrc + k; in reorderWeight()
H A DConvInt8Winograd.hpp39 …radUnit(const Convolution2D *convOp, const Tensor *input, const Tensor* weightSrc, const Tensor *o…
H A DWinogradInt8Helper.cpp319 std::shared_ptr<Tensor> WinogradInt8Helper::allocTransformWeight(const Tensor* weightSrc) { in allocTransformWeight() argument
326 bool WinogradInt8Helper::transformWeight(const Tensor* weightSrc, Tensor* weightDst) { in transformWeight() argument
359 auto dataSrcZ = weightSrc->host<int8_t>() + (ic * oz + sz) * kernelY * kernelX; in transformWeight()
H A DConvInt8Winograd.cpp543 …radUnit(const Convolution2D *convOp, const Tensor *input, const Tensor* weightSrc, const Tensor *o… in bestWinogradUnit() argument
583 support &= (extractWeight(weightSrc, weight, common, unitAttr, bn) == NO_ERROR); in bestWinogradUnit()
/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/
H A DCPUConvolution.cpp153 const int8_t* weightSrc = nullptr; in makeResourceInt8() local
155 … if (!ConvolutionCommon::getConvInt8Parameters(convParam, quanCommon, weightSrc, scalePtr, biasPtr, in makeResourceInt8()
172 temp += int(weightSrc[offset + j]); in makeResourceInt8()
187 memcpy(weightDst, weightSrc, resource->mWeightInt8->size()); in makeResourceInt8()
H A DOneDNNConvInt8.cpp61 auto weightSrc = convParam->symmetricQuan()->weight()->data(); in create() local
72 weightSrc = quanCommon->weight.get(); in create()
74 auto user_weights = memory(user_weights_md, eng, (int8_t*)weightSrc); in create()
H A DOneDNNConvolution.cpp57 const auto* weightSrc = originWeight; in OneDNNConvolution() local
64 auto user_weights = memory(user_weights_md, eng, (float*)weightSrc); in OneDNNConvolution()
/dports/emulators/ppsspp-qt5/ppsspp-1.12.3/GPU/Common/
H A DVertexDecoderX86.cpp525 X64Reg weightSrc = j < 4 ? XMM8 : XMM9; in Jit_WeightsU8Skin() local
528 weight = weightSrc; in Jit_WeightsU8Skin()
639 X64Reg weightSrc = j < 4 ? XMM8 : XMM9; in Jit_WeightsU16Skin() local
642 weight = weightSrc; in Jit_WeightsU16Skin()
/dports/emulators/ppsspp/ppsspp-1.12.3/GPU/Common/
H A DVertexDecoderX86.cpp525 X64Reg weightSrc = j < 4 ? XMM8 : XMM9; in Jit_WeightsU8Skin() local
528 weight = weightSrc; in Jit_WeightsU8Skin()
639 X64Reg weightSrc = j < 4 ? XMM8 : XMM9; in Jit_WeightsU16Skin() local
642 weight = weightSrc; in Jit_WeightsU16Skin()
/dports/emulators/libretro-ppsspp/ppsspp-1.12.3/GPU/Common/
H A DVertexDecoderX86.cpp525 X64Reg weightSrc = j < 4 ? XMM8 : XMM9; in Jit_WeightsU8Skin() local
528 weight = weightSrc; in Jit_WeightsU8Skin()
639 X64Reg weightSrc = j < 4 ? XMM8 : XMM9; in Jit_WeightsU16Skin() local
642 weight = weightSrc; in Jit_WeightsU16Skin()