Searched refs:outDepth (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| D | BoundingBox.cpp | 56 auto outDepth = op->inputs[0]->dimensions[useNchw ? 1 : 3]; in roiConstructor() local
60 op->outputs[0]->dimensions = {outBatch, outDepth, outHeight, outWidth}; in roiConstructor()
62 op->outputs[0]->dimensions = {outBatch, outHeight, outWidth, outDepth}; in roiConstructor()
|
| D | Reshape.cpp | 40 auto outDepth = op->inputs[0]->dimensions[depthIndex] * (blockSize * blockSize); in spaceToDepthConstructor() local
43 op->outputs[0]->dimensions = {op->inputs[0]->dimensions[0], outDepth, outHeight, outWidth}; in spaceToDepthConstructor()
45 op->outputs[0]->dimensions = {op->inputs[0]->dimensions[0], outHeight, outWidth, outDepth}; in spaceToDepthConstructor()
95 auto outDepth = op->inputs[0]->dimensions[depthIndex].exactDiv(blockSize * blockSize); in depthToSpaceConstructor() local
98 op->outputs[0]->dimensions = {op->inputs[0]->dimensions[0], outDepth, outHeight, outWidth}; in depthToSpaceConstructor()
100 op->outputs[0]->dimensions = {op->inputs[0]->dimensions[0], outHeight, outWidth, outDepth}; in depthToSpaceConstructor()
|