Home
last modified time | relevance | path

Searched refs:numOutputElements (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp301 int32_t numOutputElements = 1; in reshapePrepare() local
309 numOutputElements *= value; in reshapePrepare()
314 int32_t strechValue = numInputElements / numOutputElements; in reshapePrepare()
316 numOutputElements *= strechValue; in reshapePrepare()
319 NN_OPS_CHECK(numInputElements == numOutputElements); in reshapePrepare()
/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/
DReshape.cpp142 RandomVariable numOutputElements = 1; in reshapeConstructor() local
146 numOutputElements = numOutputElements * op->inputs[1]->randomBuffer[i]; in reshapeConstructor()
148 numInputElements.setEqual(numOutputElements); in reshapeConstructor()
/packages/modules/NeuralNetworks/common/cpu_operations/
DBroadcast.cpp326 uint32_t numOutputElements = getNumberOfElements(shapeOut); in subFloat32() local
327 for (uint32_t i = 0; i < numOutputElements; i++) { in subFloat32()