Searched refs:outputDimensions (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 1048 uint32_t outputDimensions[3] = {2, 2, 2}; in argMinMaxTest() local
1052 .dimensions = outputDimensions, in argMinMaxTest()
1107 uint32_t outputDimensions[5] = {2, 2, 2, 2, 2}; in expandDimsTest() local
1108 ANeuralNetworksOperandType output = getOpType(inputOperandType, 5, outputDimensions); in expandDimsTest()
1135 uint32_t outputDimensions[7] = {2, 2, 2, 2, 2, 2, 2}; in gatherTest() local
1136 ANeuralNetworksOperandType output = getOpType(inputOperandType, 7, outputDimensions); in gatherTest()
1241 uint32_t outputDimensions[2] = {2, 2}; in splitTest() local
1242 ANeuralNetworksOperandType output0 = getOpType(inputOperandType, 2, outputDimensions); in splitTest()
1243 ANeuralNetworksOperandType output1 = getOpType(inputOperandType, 2, outputDimensions); in splitTest()
1266 uint32_t outputDimensions[8] = {2, 2, 2, 2, 2, 2, 2, 2}; in tileTest() local
[all …]
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | ExecutionBuilder.h | 381 const Dimensions* outputDimensions) { in mapOutputToInput() argument
383 outputDimensions); in mapOutputToInput()
|