Lines Matching refs:useNchw

951                                      bool useNchw, std::vector<float>* scoresOutData,  in generateProposalsFloat32Compute()  argument
954 InputWithLayout<float> score_nhwc(useNchw), delta_nhwc(useNchw); in generateProposalsFloat32Compute()
969 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat32() argument
976 iouThreshold, minSize, useNchw, &scoresOut_float32, &roiOut_float32, &batchesOut)); in generateProposalsFloat32()
1012 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat16() argument
1027 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsFloat16()
1060 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsQuant() argument
1077 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsQuant()
1165 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
1181 uint32_t height = getSizeOfDimension(scoreShape, useNchw ? 2 : 1); in prepare()
1182 uint32_t width = getSizeOfDimension(scoreShape, useNchw ? 3 : 2); in prepare()
1183 uint32_t numAnchors = getSizeOfDimension(scoreShape, useNchw ? 1 : 3); in prepare()
1186 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, useNchw ? 2 : 1), height); in prepare()
1187 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, useNchw ? 3 : 2), width); in prepare()
1188 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, useNchw ? 1 : 3), numAnchors * kRoiDim); in prepare()