Searched refs:convCommon (Results 1 – 8 of 8) sorted by relevance
20 const auto convCommon = convParam->common(); in create() local21 const auto kw = convCommon->kernelX(); in create()22 const auto kh = convCommon->kernelY(); in create()23 const auto ic = convCommon->inputCount(); in create()24 const auto oc = convCommon->outputCount(); in create()25 const auto strideX = convCommon->strideX(); in create()26 const auto strideY = convCommon->strideY(); in create()35 if (convCommon->relu() || convCommon->relu6()) { in create()46 …onv_src_tz = {1, ic, convCommon->strideY() + (kh - 1) * convCommon->dilateY() + 1, (kw - 1) * conv… in create()81 return new OneDNNConvInt8(resource, convCommon, backend); in create()[all …]
16 const auto convCommon = common; in OneDNNConvolution() local17 const auto kw = convCommon->kernelX(); in OneDNNConvolution()18 const auto kh = convCommon->kernelY(); in OneDNNConvolution()19 auto ic = convCommon->inputCount(); in OneDNNConvolution()20 const auto oc = convCommon->outputCount(); in OneDNNConvolution()21 const auto strideX = convCommon->strideX(); in OneDNNConvolution()22 const auto strideY = convCommon->strideY(); in OneDNNConvolution()82 const auto convCommon = mCommon; in onResize() local83 const auto kw = convCommon->kernelX(); in onResize()84 const auto kh = convCommon->kernelY(); in onResize()[all …]
131 const auto convCommon = convParam->common(); in makeResourceInt8() local132 const auto kernelCount = convCommon->kernelY() * convCommon->kernelX(); in makeResourceInt8()134 const auto srcCount = convCommon->inputCount(); in makeResourceInt8()135 const auto outputCount = convCommon->outputCount(); in makeResourceInt8()193 resource->mRelu = convCommon->relu() || convCommon->relu6(); in makeResourceInt8()
61 parameters.kernelSize[0] = convCommon->kernelX(); in onResize()62 parameters.kernelSize[1] = convCommon->kernelY(); in onResize()63 parameters.stride[0] = convCommon->strideX(); in onResize()64 parameters.stride[1] = convCommon->strideY(); in onResize()65 parameters.dilate[0] = convCommon->dilateX(); in onResize()66 parameters.dilate[1] = convCommon->dilateY(); in onResize()74 parameters.activationType = convCommon->relu() ? 1 : (convCommon->relu6() ? 2 : 0); in onResize()244 parameters.stride[0] = convCommon->strideX(); in onResize()245 parameters.stride[1] = convCommon->strideY(); in onResize()246 parameters.dilate[0] = convCommon->dilateX(); in onResize()[all …]
69 auto& convCommon = param->common; in initialize() local70 … const int weightReduceStride = convCommon->kernelX * convCommon->kernelY * convCommon->inputCount; in initialize()71 const int oc = convCommon->outputCount / convCommon->group; in initialize()74 param->bias.resize(convCommon->outputCount); in initialize()
17 const auto& convCommon = convolutionOp->main.AsConvolution2D()->common; in merge2Convolution() local18 if (convCommon->relu || convCommon->relu6 || convolutionOp->inputIndexes.size() > 1) { in merge2Convolution()69 const auto& convCommon = convolutionOp->main.AsConvolution3D()->common; in merge2Convolution3D() local70 if (convCommon->relu || convCommon->relu6) { in merge2Convolution3D()
17 const auto& convCommon = convolutionOp->main.AsConvolution2D()->common; in merge2Convolution() local18 if (convCommon->relu || convCommon->relu6 || convolutionOp->inputIndexes.size() > 1) { in merge2Convolution()81 const auto& convCommon = convolutionOp->main.AsConvolution3D()->common; in merge2Convolution3D() local82 if (convCommon->relu || convCommon->relu6) { in merge2Convolution3D()
163 auto convCommon = mCommon; in onResize() local164 const auto kernelCount = convCommon->kernelX() * convCommon->kernelY(); in onResize()168 mIm2ColParamter.dilateX = convCommon->dilateX(); in onResize()169 mIm2ColParamter.dilateY = convCommon->dilateY(); in onResize()170 mIm2ColParamter.strideX = convCommon->strideX(); in onResize()171 mIm2ColParamter.strideY = convCommon->strideY(); in onResize()172 mIm2ColParamter.padX = convCommon->padX(); in onResize()173 mIm2ColParamter.padY = convCommon->padY(); in onResize()175 mIm2ColParamter.kernelX = convCommon->kernelX(); in onResize()176 mIm2ColParamter.kernelY = convCommon->kernelY(); in onResize()