Searched refs:batchId (Results 1 – 2 of 2) sorted by relevance
81 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignNhwc() local86 NN_RET_CHECK_GE(batchId, 0); in roiAlignNhwc()87 NN_RET_CHECK_LT(batchId, numBatches); in roiAlignNhwc()119 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignNhwc()208 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignQuantNhwc() local218 NN_RET_CHECK_GE(batchId, 0); in roiAlignQuantNhwc()219 NN_RET_CHECK_LT(batchId, numBatches); in roiAlignQuantNhwc()248 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignQuantNhwc()
77 uint32_t batchId = batchSplitData[roiIndex]; in roiPoolingNhwc() local82 NN_RET_CHECK_GE(batchId, 0); in roiPoolingNhwc()83 NN_RET_CHECK_LT(batchId, numBatches); in roiPoolingNhwc()106 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiPoolingNhwc()