| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RoiPooling.cpp | 44 const Shape& /*batchSplitShape*/, float heightStride, float widthStride, in roiPoolingNhwc() argument
50 const T_Roi widthScale = 1.0f / widthStride; in roiPoolingNhwc()
133 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPooling() argument
140 batchSplitData, batchSplitShape, heightStride, widthStride, in roiPooling()
151 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling() argument
156 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
166 float widthStride, bool useNchw, int8_t* outputData, in roiPooling() argument
171 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
196 float heightStride, widthStride; in prepare() local
199 widthStride = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
[all …]
|
| D | RoiAlign.cpp | 49 const Shape& /*batchSplitShape*/, float heightStride, float widthStride, in roiAlignNhwc() argument
56 const T_Roi widthScale = 1.0f / widthStride; in roiAlignNhwc()
175 float heightStride, float widthStride, int32_t heightSamplingRatio, in roiAlignQuantNhwc() argument
183 const float widthScale = 1.0f / widthStride; in roiAlignQuantNhwc()
307 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlign() argument
318 batchSplitShape, heightStride, widthStride, heightSamplingRatio, widthSamplingRatio, in roiAlign()
322 batchSplitData, batchSplitShape, heightStride, widthStride, in roiAlign()
|
| D | GenerateProposals.cpp | 850 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsNhwcFloat32Compute() argument
876 float wShift = w * widthStride; in generateProposalsNhwcFloat32Compute()
952 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsFloat32Compute() argument
963 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, in generateProposalsFloat32Compute()
971 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsFloat32() argument
978 imageInfoData, imageInfoShape, heightStride, widthStride, preNmsTopN, postNmsTopN, in generateProposalsFloat32()
1014 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsFloat16() argument
1030 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsFloat16()
1063 float heightStride, float widthStride, int32_t preNmsTopN, in generateProposalsQuant() argument
1082 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsQuant()
|