| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RoiPooling.cpp | 44 const Shape& /*batchSplitShape*/, float heightStride, float widthStride, in roiPoolingNhwc() argument
49 const T_Roi heightScale = 1.0f / heightStride; in roiPoolingNhwc()
133 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPooling() argument
140 batchSplitData, batchSplitShape, heightStride, widthStride, in roiPooling()
150 const Shape& batchSplitShape, float heightStride, in roiPooling() argument
156 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
165 const Shape& batchSplitShape, float heightStride, in roiPooling() argument
171 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
196 float heightStride, widthStride; in prepare() local
198 heightStride = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
[all …]
|
| D | RoiAlign.cpp | 49 const Shape& /*batchSplitShape*/, float heightStride, float widthStride, in roiAlignNhwc() argument
55 const T_Roi heightScale = 1.0f / heightStride; in roiAlignNhwc()
175 float heightStride, float widthStride, int32_t heightSamplingRatio, in roiAlignQuantNhwc() argument
182 const float heightScale = 1.0f / heightStride; 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
873 float hShift = h * heightStride; 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()
|