Searched refs:mSubConvolution (Results 1 – 2 of 2) sorted by relevance
17 mSubConvolution = subConvolution; in ConvolutionGroup()38 mInputUnit->buffer().dim[1].extent = ib.dim[1].extent / mSubConvolution.size(); in onResize()49 mOutputUnit->buffer().dim[1].extent = ob.dim[1].extent / mSubConvolution.size(); in onResize()59 for (auto &iter : mSubConvolution) { in onResize()84 …nt inputGroupSize = input->width() * input->height() * input->channel() / mSubConvolution.size(); in onExecute()85 …outputGroupSize = output->width() * output->height() * output->channel() / mSubConvolution.size(); in onExecute()86 int subInputChannel = input->channel() / mSubConvolution.size(); in onExecute()87 int subOutputChannel = output->channel() / mSubConvolution.size(); in onExecute()88 for (int group = 0; group < mSubConvolution.size(); ++group) { in onExecute()91 mSubConvolution[group]->onExecute(mInputUnitWrap, mOutputUnitWrap); in onExecute()
33 std::vector<std::shared_ptr<Execution>> mSubConvolution; member in MNN::ConvolutionGroup