提交 d52fd00a 编写于 作者: S sweetsky0901

int to size_t

上级 a8109cf0
...@@ -63,7 +63,7 @@ class DetectionOutputKernel : public framework::OpKernel<T> { ...@@ -63,7 +63,7 @@ class DetectionOutputKernel : public framework::OpKernel<T> {
float nms_threshold = context.template Attr<float>("nms_threshold"); float nms_threshold = context.template Attr<float>("nms_threshold");
float confidence_threshold = float confidence_threshold =
context.template Attr<float>("confidence_threshold"); context.template Attr<float>("confidence_threshold");
int batch_size = in_conf->dims()[1]; size_t batch_size = in_conf->dims()[1];
int conf_sum_size = in_conf->numel(); int conf_sum_size = in_conf->numel();
// for softmax // for softmax
std::vector<int64_t> conf_shape_softmax_vec( std::vector<int64_t> conf_shape_softmax_vec(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册