Searched refs:boxesBoxDim (Results 1 – 3 of 3) sorted by relevance
2002 size_t boxesBoxDim = boxes.dims().size() - 2; in verify() local2009 boxesDims[boxesBoxDim], scoresDims[scoresBoxDim], this, in verify()2021 boxesDims[boxesBoxDim], scoresDims[scoresBoxDim], this, in verify()
1262 int boxesBoxDim = boxTensorDimSize - 2; in libjit_nms_generic() local1266 size_t numBoxes = boxTensorDims[boxesBoxDim]; in libjit_nms_generic()1276 assert(scoresTensorDims[scoresBoxDim] == boxTensorDims[boxesBoxDim] && in libjit_nms_generic()1284 numBoxes = boxTensorDims[boxesBoxDim]; in libjit_nms_generic()
5287 int boxesBoxDim = boxes->dims().size() - 2; in fwdNonMaxSuppressionInstImpl() local5291 dim_t numBoxes = boxes->dims()[boxesBoxDim]; in fwdNonMaxSuppressionInstImpl()5301 assert(scores->dims()[scoresBoxDim] == boxes->dims()[boxesBoxDim] && in fwdNonMaxSuppressionInstImpl()5309 numBoxes = boxes->dims()[boxesBoxDim]; in fwdNonMaxSuppressionInstImpl()