Home
last modified time | relevance | path

Searched refs:batchBase (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DRoiAlign.cpp109 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignNhwc() local
154 interpolation += ws[c] * batchBase[offsets[c] + k]; in roiAlignNhwc()
238 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignQuantNhwc() local
282 wQuant * (static_cast<int32_t>(batchBase[offsets[c] + k]) - in roiAlignQuantNhwc()
DRoiPooling.cpp94 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiPoolingNhwc() local
114 T_Input inputValue = batchBase[h * inWidth * inDepth + w * inDepth + k]; in roiPoolingNhwc()