| /dports/misc/mnn/MNN-1.2.0/tools/train/source/demo/ |
| H A D | nnGradTest.cpp | 28 int weightSize = ic * oc * kw * kh; in run() local
90 int weightSize = ic * oc * kw * kh; in run() local
161 int weightSize = ic * oc * kw * kh; in run() local
186 int weightSize = oc * kw * kh; in run() local
240 int weightSize = l * h; in run() local
276 int weightSize = b * l * h; in run() local
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/vulkan/execution/ |
| H A D | VulkanConvolution.cpp | 26 … const float* weightSource, size_t weightSize) { in _createBufferForConvDepthwise()
111 bool VulkanConvolutionDepthwise::_init(const float* weightData, size_t weightSize, const Op* convOp… in _init()
154 VulkanConvolutionDepthwise::VulkanConvolutionDepthwise(const float* weightData, size_t weightSize, … in VulkanConvolutionDepthwise()
243 int weightSize = 0; in onCreate() local
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/coreml/execution/ |
| H A D | CoreMLConvolution.hpp | 29 int weightSize, biasSize; member in MNN::CoreMLConvolution
|
| /dports/math/mlpack/mlpack-3.4.2/src/mlpack/tests/ |
| H A D | ann_visitor_test.cpp | 61 size_t weightSize = boost::apply_visitor(WeightSizeVisitor(), in CheckCorrectnessOfWeightSize() local
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/tensorrt/execution/ |
| H A D | TRTDeconvolution.cpp | 35 int weightSize = 0; in onEncode() local
|
| H A D | TRTDepthwiseDeconvolution.cpp | 35 int weightSize = 0; in onEncode() local
|
| H A D | TRTDepthwiseConvolution.cpp | 36 int weightSize = 0; in onEncode() local
|
| H A D | TRTConvolution.cpp | 34 int weightSize = 0; in onEncode() local
|
| /dports/games/blokish/blokish/src/ |
| H A D | ai.h | 42 int weightSize; member
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/hiai/execution/ |
| H A D | NPUDeconvolution.cpp | 34 int weightSize = 0; in onResize() local
|
| H A D | NPUConvolution.cpp | 34 int weightSize = 0; in onResize() local
|
| H A D | NPUConvolutionDepthwise.cpp | 34 int weightSize = 0; in onResize() local
|
| H A D | NPUConvolutionDepthwiseInt8.cpp | 34 int weightSize = quantizedParams->weight()->size(); in onResize() local
|
| H A D | NPUConvolutionInt8.cpp | 34 int weightSize = quantizedParams->weight()->size(); in onResize() local
|
| /dports/misc/mnn/MNN-1.2.0/tools/converter/source/tflite/ |
| H A D | ConvolutionTflite.cpp | 47 const int weightSize = co * kh * kw * ci; in run() local
259 const int weightSize = co * kh * kw * ci; in run() local
|
| H A D | DepthwiseConv2DTflite.cpp | 46 const int weightSize = kh * kw * ci; in run() local
|
| /dports/misc/mnn/MNN-1.2.0/tools/cpp/ |
| H A D | IDSTEncoder.hpp | 98 static float GetSparsity(const float* weightData, int weightSize, unsigned int& nnz, const float* a… in GetSparsity()
158 static unsigned int GetBestMaxStep(const float* weightData, int weightSize, unsigned char& iMaxStep… in GetBestMaxStep()
266 int weightSize = area * channel; in WriteSparseQuanBlobs() local
|
| /dports/misc/mnn/MNN-1.2.0/tools/converter/source/onnx/ |
| H A D | MatMulOnnx.cpp | 56 int weightSize = 1; in run() local
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/image/ |
| H A D | Conv2DBackPropFilter.cpp | 43 const int weightSize = inputs[0]->elementSize(); in onResize() local
|
| /dports/misc/mnn/MNN-1.2.0/tools/converter/source/common/ |
| H A D | writeFb.cpp | 122 const int weightSize = param->weight.size(); in writeFb() local
194 int weightSize = param->weight.size(); in writeFb() local
459 int weightSize = param->weight.size(); in writeFb() local
|
| /dports/misc/mnn/MNN-1.2.0/pymnn/pip_package/MNN/tools/mnn_fb/ |
| H A D | LSTM.py | 106 def LSTMAddWeightSize(builder, weightSize): builder.PrependInt32Slot(1, weightSize, 0) argument
|
| H A D | InnerProduct.py | 114 def InnerProductAddWeightSize(builder, weightSize): builder.PrependInt32Slot(2, weightSize, 0) argument
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/buffer/ |
| H A D | ConvBufExecution.cpp | 317 int weightSize = 0; in ConvBufExecution() local
475 int weightSize = 0; in onResize() local
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/cuda/execution/ |
| H A D | DeconvSingleInputExecution.cu | 81 int weightSize = 0; in DeconvSingleInputExecution() local
|
| H A D | ConvSingleInputExecution.cu | 86 int weightSize = 0; in ConvSingleInputExecution() local
|