Searched refs:outputDimensions (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 1037 uint32_t outputDimensions[3] = {2, 2, 2}; in argMinMaxTest() local
1041 .dimensions = outputDimensions, in argMinMaxTest()
1096 uint32_t outputDimensions[5] = {2, 2, 2, 2, 2}; in expandDimsTest() local
1097 ANeuralNetworksOperandType output = getOpType(inputOperandType, 5, outputDimensions); in expandDimsTest()
1124 uint32_t outputDimensions[7] = {2, 2, 2, 2, 2, 2, 2}; in gatherTest() local
1125 ANeuralNetworksOperandType output = getOpType(inputOperandType, 7, outputDimensions); in gatherTest()
1230 uint32_t outputDimensions[2] = {2, 2}; in splitTest() local
1231 ANeuralNetworksOperandType output0 = getOpType(inputOperandType, 2, outputDimensions); in splitTest()
1232 ANeuralNetworksOperandType output1 = getOpType(inputOperandType, 2, outputDimensions); in splitTest()
1255 uint32_t outputDimensions[8] = {2, 2, 2, 2, 2, 2, 2, 2}; in tileTest() local
[all …]
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | ExecutionBuilder.h | 359 const Dimensions* outputDimensions) { in mapOutputToInput() argument
361 outputDimensions); in mapOutputToInput()
|