Searched refs:weightPartSize (Results 1 – 2 of 2) sorted by relevance
40 int weightPartSize = conv2D->weight.size() / outputCount; in merge2Convolution() local57 for (int j = 0; j < weightPartSize; ++j) { in merge2Convolution()58 conv2D->weight[i * weightPartSize + j] *= a; in merge2Convolution()87 int weightPartSize = conv3D->weight.size() / outputCount; in merge2Convolution3D() local90 for (int j = 0; j < weightPartSize; ++j) { in merge2Convolution3D()91 conv3D->weight[i * weightPartSize + j] *= a; in merge2Convolution3D()
52 int weightPartSize = conv2D->weight.size() / outputCount; in merge2Convolution() local69 for (int j = 0; j < weightPartSize; ++j) { in merge2Convolution()70 conv2D->weight[i * weightPartSize + j] *= a; in merge2Convolution()111 int weightPartSize = conv3D->weight.size() / outputCount; in merge2Convolution3D() local114 for (int j = 0; j < weightPartSize; ++j) { in merge2Convolution3D()115 conv3D->weight[i * weightPartSize + j] *= a; in merge2Convolution3D()