Searched defs:outDepth (Results 1 – 2 of 2) sorted by relevance
56 auto outDepth = op->inputs[0]->dimensions[useNchw ? 1 : 3]; in roiConstructor() local
40 auto outDepth = op->inputs[0]->dimensions[depthIndex] * (blockSize * blockSize); in spaceToDepthConstructor() local95 auto outDepth = op->inputs[0]->dimensions[depthIndex].exactDiv(blockSize * blockSize); in depthToSpaceConstructor() local