Lines Matching refs:inputShape
55 inline bool roiPoolingNhwc(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPoolingNhwc() argument
65 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiPoolingNhwc()
66 uint32_t inHeight = getSizeOfDimension(inputShape, 1); in roiPoolingNhwc()
67 uint32_t inWidth = getSizeOfDimension(inputShape, 2); in roiPoolingNhwc()
68 uint32_t inDepth = getSizeOfDimension(inputShape, 3); in roiPoolingNhwc()
123 T_Input maxValue = static_cast<T_Input>(inputShape.offset); in roiPoolingNhwc()
144 inline bool roiPooling(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPooling() argument
150 NN_RET_CHECK(input.initialize(inputData, inputShape)); in roiPooling()
160 inline bool roiPooling<uint8_t, uint16_t>(const uint8_t* inputData, const Shape& inputShape,
168 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData,
175 inline bool roiPooling<int8_t, uint16_t>(const int8_t* inputData, const Shape& inputShape,
183 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData,