Searched refs:conv3D (Results 1 – 3 of 3) sorted by relevance
81 auto conv3D = convolutionOp->main.AsConvolution3D(); in merge2Convolution3D() local82 int outputCount = conv3D->common->outputCount; in merge2Convolution3D()84 conv3D->bias[i] = conv3D->bias[i] * alpha[i] + bias[i]; in merge2Convolution3D()87 int weightPartSize = conv3D->weight.size() / outputCount; in merge2Convolution3D()91 conv3D->weight[i * weightPartSize + j] *= a; in merge2Convolution3D()
105 auto conv3D = convolutionOp->main.AsConvolution3D(); in merge2Convolution3D() local106 int outputCount = conv3D->common->outputCount; in merge2Convolution3D()108 conv3D->bias[i] = conv3D->bias[i] * alpha[i] + bias[i]; in merge2Convolution3D()111 int weightPartSize = conv3D->weight.size() / outputCount; in merge2Convolution3D()115 conv3D->weight[i * weightPartSize + j] *= a; in merge2Convolution3D()
1 …conv3D: Either strides or dilations must be 1. Got strides \"+n+\" and dilations '\"+a+\"'\"})),C(…