Searched refs:outBatch (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| D | BoundingBox.cpp | 55 auto outBatch = op->inputs[1]->dimensions[0]; 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 | 181 auto outBatch = op->inputs[0]->dimensions[0].exactDiv(blockHeight * blockWidth); in batchToSpaceConstructor() local
186 op->outputs[0]->dimensions = {outBatch, op->inputs[0]->dimensions[1], outHeight, outWidth}; in batchToSpaceConstructor()
188 op->outputs[0]->dimensions = {outBatch, outHeight, outWidth, op->inputs[0]->dimensions[3]}; in batchToSpaceConstructor()
247 auto outBatch = op->inputs[0]->dimensions[0] * (blockHeight * blockWidth); in spaceToBatchConstructor() local
252 op->outputs[0]->dimensions = {outBatch, op->inputs[0]->dimensions[1], outHeight, outWidth}; in spaceToBatchConstructor()
254 op->outputs[0]->dimensions = {outBatch, outHeight, outWidth, op->inputs[0]->dimensions[3]}; in spaceToBatchConstructor()
|