Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DOperationsUtils.cpp394 int32_t numOutputElements = 1; in reshapePrepare() local
402 numOutputElements *= value; in reshapePrepare()
407 int32_t strechValue = numInputElements / numOutputElements; in reshapePrepare()
409 numOutputElements *= strechValue; in reshapePrepare()
412 NN_OPS_CHECK(numInputElements == numOutputElements); in reshapePrepare()
/frameworks/ml/nn/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()
/frameworks/ml/nn/common/operations/
DBroadcast.cpp323 uint32_t numOutputElements = getNumberOfElements(shapeOut); in subFloat32() local
324 for (uint32_t i = 0; i < numOutputElements; i++) { in subFloat32()