| /frameworks/ml/nn/common/operations/ |
| D | RoiPooling.cpp | 56 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPoolingNhwc() argument
62 const T_Roi widthScale = 1.0f / widthStride; in roiPoolingNhwc()
145 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPooling() argument
152 batchSplitData, batchSplitShape, heightStride, widthStride, in roiPooling()
163 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling() argument
168 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
178 float widthStride, bool useNchw, int8_t* outputData, in roiPooling() argument
183 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
247 float heightStride, widthStride; in prepare() local
250 widthStride = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
[all …]
|
| D | RoiAlign.cpp | 59 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlignNhwc() argument
66 const T_Roi widthScale = 1.0f / widthStride; in roiAlignNhwc()
185 float heightStride, float widthStride, int32_t heightSamplingRatio, in roiAlignQuantNhwc() argument
193 const float widthScale = 1.0f / widthStride; in roiAlignQuantNhwc()
317 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlign() argument
328 batchSplitShape, heightStride, widthStride, heightSamplingRatio, widthSamplingRatio, in roiAlign()
332 batchSplitData, batchSplitShape, heightStride, widthStride, in roiAlign()
|
| D | GenerateProposals.cpp | 956 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsNhwcFloat32Compute() argument
982 float wShift = w * widthStride; in generateProposalsNhwcFloat32Compute()
1058 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsFloat32Compute() argument
1069 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, in generateProposalsFloat32Compute()
1077 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsFloat32() argument
1084 imageInfoData, imageInfoShape, heightStride, widthStride, preNmsTopN, postNmsTopN, in generateProposalsFloat32()
1120 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsFloat16() argument
1136 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsFloat16()
1169 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsQuant() argument
1188 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsQuant()
|