Searched refs:batchSplitDim (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
3195 getOpType(ANEURALNETWORKS_TENSOR_INT32, 1, batchSplitDim), in roiAlignOpTest()
3225 uint32_t inDim[] = {1, 4, 4, 1}, roiDim[] = {4, 4}, batchSplitDim[] = {1}; in roiPoolingOpTest() local
3230 getOpType(ANEURALNETWORKS_TENSOR_INT32, 1, batchSplitDim), in roiPoolingOpTest()
|