Searched refs:depthIndex (Results 1 – 1 of 1) sorted by relevance
33 int depthIndex = useNchw ? 1 : 3; in spaceToDepthConstructor() local40 auto outDepth = op->inputs[0]->dimensions[depthIndex] * (blockSize * blockSize); in spaceToDepthConstructor()88 int depthIndex = useNchw ? 1 : 3; in depthToSpaceConstructor() local95 auto outDepth = op->inputs[0]->dimensions[depthIndex].exactDiv(blockSize * blockSize); in depthToSpaceConstructor()