Searched refs:originGEqual (Results 1 – 1 of 1) sorted by relevance
68 std::vector<std::shared_ptr<Tensor>> originGEqual; in onComputeMaxPool() local69 originGEqual.resize(kernel_h * kernel_w); in onComputeMaxPool()159 originGEqual[index].reset(new Tensor); in onComputeMaxPool()161 originGEqual[index]->buffer().dimensions = 4; in onComputeMaxPool()162 originGEqual[index]->setLength(0, ob); in onComputeMaxPool()163 originGEqual[index]->setLength(1, oc); in onComputeMaxPool()164 originGEqual[index]->setLength(2, oh); in onComputeMaxPool()165 originGEqual[index]->setLength(3, ow); in onComputeMaxPool()166 des = TensorUtils::getDescribe(originGEqual[index].get()); in onComputeMaxPool()185 cmd1.outputs = {originGEqual[index].get()}; in onComputeMaxPool()[all …]