Searched refs:outputDimensions (Results 1 – 1 of 1) sorted by relevance
/frameworks/ml/nn/runtime/test/ |
D | TestValidateOperations.cpp | 1038 uint32_t outputDimensions[3] = {2, 2, 2}; in argMinMaxTest() local 1042 .dimensions = outputDimensions, in argMinMaxTest() 1097 uint32_t outputDimensions[5] = {2, 2, 2, 2, 2}; in expandDimsTest() local 1098 ANeuralNetworksOperandType output = getOpType(inputOperandType, 5, outputDimensions); in expandDimsTest() 1125 uint32_t outputDimensions[7] = {2, 2, 2, 2, 2, 2, 2}; in gatherTest() local 1126 ANeuralNetworksOperandType output = getOpType(inputOperandType, 7, outputDimensions); in gatherTest() 1231 uint32_t outputDimensions[2] = {2, 2}; in splitTest() local 1232 ANeuralNetworksOperandType output0 = getOpType(inputOperandType, 2, outputDimensions); in splitTest() 1233 ANeuralNetworksOperandType output1 = getOpType(inputOperandType, 2, outputDimensions); in splitTest() 1256 uint32_t outputDimensions[8] = {2, 2, 2, 2, 2, 2, 2, 2}; in tileTest() local [all …]
|