Home
last modified time | relevance | path

Searched refs:numAxis (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/
DReduce.cpp30 uint32_t numAxis = getUniform<int32_t>(1, 10); in reduceOpConstructor() local
31 op->inputs[1]->dimensions = {numAxis}; in reduceOpConstructor()
32 op->inputs[1]->resizeBuffer<int32_t>(numAxis); in reduceOpConstructor()
33 for (uint32_t i = 0; i < numAxis; i++) { in reduceOpConstructor()
DReshape.cpp454 uint32_t numAxis = getUniform<int32_t>(1, 10); in squeezeConstructor() local
455 op->inputs[1]->dimensions = {numAxis}; in squeezeConstructor()
456 op->inputs[1]->resizeBuffer<int32_t>(numAxis); in squeezeConstructor()
457 for (uint32_t i = 0; i < numAxis; i++) { in squeezeConstructor()