Searched refs:batchBase (Results 1 – 2 of 2) sorted by relevance
119 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignNhwc() local164 interpolation += ws[c] * batchBase[offsets[c] + k]; in roiAlignNhwc()248 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignQuantNhwc() local292 wQuant * (static_cast<int32_t>(batchBase[offsets[c] + k]) - in roiAlignQuantNhwc()
106 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiPoolingNhwc() local126 T_Input inputValue = batchBase[h * inWidth * inDepth + w * inDepth + k]; in roiPoolingNhwc()