Home
last modified time | relevance | path

Searched defs:outBatch (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/
DReshape.cpp181 auto outBatch = op->inputs[0]->dimensions[0].exactDiv(blockHeight * blockWidth); in batchToSpaceConstructor() local
247 auto outBatch = op->inputs[0]->dimensions[0] * (blockHeight * blockWidth); in spaceToBatchConstructor() local
DBoundingBox.cpp55 auto outBatch = op->inputs[1]->dimensions[0]; in roiConstructor() local
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DResizeLayer.cpp49 unsigned int outBatch = inputShape[0]; in InferOutputShapes() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DBatchToSpaceNd.cpp70 const unsigned int outBatch = inBatch % outputBatchSize; in BatchToSpaceNd() local