Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/
DReshape.cpp134 RandomVariable numOutputElements = 1; in reshapeConstructor() local
138 numOutputElements = numOutputElements * op->inputs[1]->randomBuffer[i]; in reshapeConstructor()
140 numInputElements.setEqual(numOutputElements); in reshapeConstructor()
/frameworks/ml/nn/common/
DOperationsUtils.cpp418 int32_t numOutputElements = 1; in reshapePrepare() local
426 numOutputElements *= value; in reshapePrepare()
431 int32_t strechValue = numInputElements / numOutputElements; in reshapePrepare()
433 numOutputElements *= strechValue; in reshapePrepare()
436 NN_OPS_CHECK(numInputElements == numOutputElements); in reshapePrepare()
/frameworks/ml/nn/common/operations/
DBroadcast.cpp253 uint32_t numOutputElements = getNumberOfElements(shapeOut); in subFloat32() local
254 for (uint32_t i = 0; i < numOutputElements; i++) { in subFloat32()