Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/
DOperationsUtils.cpp392 int32_t numOutputElements = 1; in reshapePrepare() local
400 numOutputElements *= value; in reshapePrepare()
405 int32_t strechValue = numInputElements / numOutputElements; in reshapePrepare()
407 numOutputElements *= strechValue; in reshapePrepare()
410 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/operations/
DBroadcast.cpp326 uint32_t numOutputElements = getNumberOfElements(shapeOut); in subFloat32() local
327 for (uint32_t i = 0; i < numOutputElements; i++) { in subFloat32()