Searched refs:inDim (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3190 uint32_t inDim[] = {1, 4, 4, 1}, roiDim[] = {4, 4}, batchSplitDim[] = {1}; in roiAlignOpTest() local
3194 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiAlignOpTest()
3225 uint32_t inDim[] = {1, 4, 4, 1}, roiDim[] = {4, 4}, batchSplitDim[] = {1}; in roiPoolingOpTest() local
3229 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiPoolingOpTest()
3313 uint32_t inDim[] = {1, 3, 3, 2}, filterDim[] = {2, 2, 2, 1}, biasDim[] = {2}; in groupedConvOpTest() local
3315 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 4, inDim); in groupedConvOpTest()
3388 uint32_t inDim[] = {1, 2, 2, 2}, filterDim[] = {2, 3, 3, 1}, biasDim[] = {2}; in transposeConvOpTest() local
3390 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 4, inDim); in transposeConvOpTest()
|