@@ -273,8 +273,8 @@ ExperimentalDetectronDetectionOutput::ExperimentalDetectronDetectionOutput(const
273273 score_threshold_ = attributes.score_threshold ;
274274 nms_threshold_ = attributes.nms_threshold ;
275275 max_delta_log_wh_ = attributes.max_delta_log_wh ;
276- classes_num_ = static_cast <int32_t >(attributes.num_classes );
277- max_detections_per_class_ = static_cast <int32_t >(attributes.post_nms_count );
276+ classes_num_ = static_cast <int >(attributes.num_classes );
277+ max_detections_per_class_ = static_cast <int >(attributes.post_nms_count );
278278 max_detections_per_image_ = attributes.max_detections_per_image ;
279279 class_agnostic_box_regression_ = attributes.class_agnostic_box_regression ;
280280 deltas_weights_ = attributes.deltas_weights ;
@@ -299,9 +299,9 @@ void ExperimentalDetectronDetectionOutput::initSupportedPrimitiveDescriptors() {
299299}
300300
301301void ExperimentalDetectronDetectionOutput::execute ([[maybe_unused]] const dnnl::stream& strm) {
302- const auto rois_num = static_cast <int32_t >(getParentEdgeAt (INPUT_ROIS )->getMemory ().getStaticDims ()[0 ]);
303- assert (classes_num_ == static_cast <int32_t >(getParentEdgeAt (INPUT_SCORES )->getMemory ().getStaticDims ()[1 ]));
304- assert (4 * classes_num_ == static_cast <int32_t >(getParentEdgeAt (INPUT_DELTAS )->getMemory ().getStaticDims ()[1 ]));
302+ const auto rois_num = static_cast <int64_t >(getParentEdgeAt (INPUT_ROIS )->getMemory ().getStaticDims ()[0 ]);
303+ assert (classes_num_ == static_cast <int64_t >(getParentEdgeAt (INPUT_SCORES )->getMemory ().getStaticDims ()[1 ]));
304+ assert (4 * classes_num_ == static_cast <int64_t >(getParentEdgeAt (INPUT_DELTAS )->getMemory ().getStaticDims ()[1 ]));
305305
306306 const auto * boxes = getSrcDataAtPortAs<const float >(INPUT_ROIS );
307307 const auto * deltas = getSrcDataAtPortAs<const float >(INPUT_DELTAS );
@@ -315,8 +315,8 @@ void ExperimentalDetectronDetectionOutput::execute([[maybe_unused]] const dnnl::
315315 std::vector<float > refined_boxes (classes_num_ * rois_num * 4 , 0 );
316316 std::vector<float > refined_scores (classes_num_ * rois_num, 0 );
317317 std::vector<float > refined_boxes_areas (classes_num_ * rois_num, 0 );
318- Indexer refined_box_idx ({classes_num_, rois_num, 4 });
319- Indexer refined_score_idx ({classes_num_, rois_num});
318+ Indexer refined_box_idx ({static_cast < int >( classes_num_), static_cast < int >( rois_num) , 4 });
319+ Indexer refined_score_idx ({static_cast < int >( classes_num_), static_cast < int >( rois_num) });
320320
321321 refine_boxes (boxes,
322322 deltas,
@@ -325,8 +325,8 @@ void ExperimentalDetectronDetectionOutput::execute([[maybe_unused]] const dnnl::
325325 refined_boxes.data (),
326326 refined_boxes_areas.data (),
327327 refined_scores.data (),
328- rois_num,
329- classes_num_,
328+ static_cast < int >( rois_num) ,
329+ static_cast < int >( classes_num_) ,
330330 max_delta_log_wh_,
331331 1 .0F );
332332
@@ -343,7 +343,7 @@ void ExperimentalDetectronDetectionOutput::execute([[maybe_unused]] const dnnl::
343343 buffer.data (),
344344 &indices[total_detections_num],
345345 detections_per_class[class_idx],
346- rois_num,
346+ static_cast < int >( rois_num) ,
347347 -1 ,
348348 static_cast <int >(max_detections_per_class_),
349349 score_threshold_,
0 commit comments