Searched refs:batchId (Results 1 – 3 of 3) sorted by relevance
71 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignNhwc() local76 NN_RET_CHECK_GE(batchId, 0u); in roiAlignNhwc()77 NN_RET_CHECK_LT(batchId, numBatches); in roiAlignNhwc()109 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignNhwc()198 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignQuantNhwc() local208 NN_RET_CHECK_GE(batchId, 0u); in roiAlignQuantNhwc()209 NN_RET_CHECK_LT(batchId, numBatches); in roiAlignQuantNhwc()238 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignQuantNhwc()
65 uint32_t batchId = batchSplitData[roiIndex]; in roiPoolingNhwc() local70 NN_RET_CHECK_GE(batchId, 0u); in roiPoolingNhwc()71 NN_RET_CHECK_LT(batchId, numBatches); in roiPoolingNhwc()94 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiPoolingNhwc()
261 String batchId = pair[0]; in parse() local263 return Pair.create(batchId, index); in parse()