Searched refs:boxesBatchDim (Results 1 – 4 of 4) sorted by relevance
2003 size_t boxesBatchDim = boxes.dims().size() - 3; in verify() local2016 "Batch dimension doesn't match.", boxesDims[boxesBatchDim], in verify()
4511 int boxesBatchDim = boxesDim.size() - 3; in createNonMaxSuppressionONNX() local4525 {boxesDim[boxesBatchDim] * static_cast<dim_t>(maxOutputBoxesPerClass)}); in createNonMaxSuppressionONNX()
1270 int boxesBatchDim = boxTensorDimSize - 3; in libjit_nms_generic() local1278 assert(scoresTensorDims[scoresBatchDim] == boxTensorDims[boxesBatchDim] && in libjit_nms_generic()1280 (void)boxesBatchDim; in libjit_nms_generic()
5296 int boxesBatchDim = boxes->dims().size() - 3; in fwdNonMaxSuppressionInstImpl() local5303 assert(scores->dims()[scoresBatchDim] == boxes->dims()[boxesBatchDim] && in fwdNonMaxSuppressionInstImpl()5305 (void)boxesBatchDim; in fwdNonMaxSuppressionInstImpl()