Home
last modified time | relevance | path

Searched defs:outputDimensions (Results 1 – 2 of 2) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/runtime/
H A DExecutionBuilder.h359 const Dimensions* outputDimensions) { in mapOutputToInput()
/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestValidateOperations.cpp1037 uint32_t outputDimensions[3] = {2, 2, 2}; in argMinMaxTest() local
1096 uint32_t outputDimensions[5] = {2, 2, 2, 2, 2}; in expandDimsTest() local
1124 uint32_t outputDimensions[7] = {2, 2, 2, 2, 2, 2, 2}; in gatherTest() local
1230 uint32_t outputDimensions[2] = {2, 2}; in splitTest() local
1255 uint32_t outputDimensions[8] = {2, 2, 2, 2, 2, 2, 2, 2}; in tileTest() local
1274 uint32_t outputDimensions[4] = {4, 5, 6, 3}; in topkV2Test() local
1663 uint32_t outputDimensions[2] = {4, 6}; in reshapeOpTest() local
1750 uint32_t outputDimensions[4] = {4, 3, 4, 3}; in padOpTest() local
1778 uint32_t outputDimensions[4] = {4, 3, 4, 3}; in padV2OpTest() local
4419 uint32_t outputDimensions[3] = {3, 4, 5}; in fillTest() local