| /frameworks/ml/nn/common/operations/ |
| D | RoiPooling.cpp | 56 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPoolingNhwc() argument
61 const T_Roi heightScale = 1.0f / heightStride; in roiPoolingNhwc()
145 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPooling() argument
152 batchSplitData, batchSplitShape, heightStride, widthStride, in roiPooling()
162 const Shape& batchSplitShape, float heightStride, in roiPooling() argument
168 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
177 const Shape& batchSplitShape, float heightStride, in roiPooling() argument
183 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
247 float heightStride, widthStride; in prepare() local
249 heightStride = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
[all …]
|
| D | RoiAlign.cpp | 59 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlignNhwc() argument
65 const T_Roi heightScale = 1.0f / heightStride; in roiAlignNhwc()
185 float heightStride, float widthStride, int32_t heightSamplingRatio, in roiAlignQuantNhwc() argument
192 const float heightScale = 1.0f / heightStride; 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
979 float hShift = h * heightStride; 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()
|