@@ -41,32 +41,33 @@ TrtShapeEstimator::TrtShapeEstimator(
41
41
const auto pc_input_dims = trt_common_->getBindingDimensions (0 );
42
42
const auto pc_input_size = std::accumulate (
43
43
pc_input_dims.d + 1 , pc_input_dims.d + pc_input_dims.nbDims , 1 , std::multiplies<int >());
44
- input_pc_d_ = cuda_utils::make_unique<float []>(pc_input_size * batch_config[2 ]);
44
+ input_pc_d_ = autoware:: cuda_utils::make_unique<float []>(pc_input_size * batch_config[2 ]);
45
45
batch_size_ = batch_config[2 ];
46
46
const auto one_hot_input_dims = trt_common_->getBindingDimensions (1 );
47
47
const auto one_hot_input_size = std::accumulate (
48
48
one_hot_input_dims.d + 1 , one_hot_input_dims.d + one_hot_input_dims.nbDims , 1 ,
49
49
std::multiplies<int >());
50
- input_one_hot_d_ = cuda_utils::make_unique<float []>(one_hot_input_size * batch_config[2 ]);
50
+ input_one_hot_d_ =
51
+ autoware::cuda_utils::make_unique<float []>(one_hot_input_size * batch_config[2 ]);
51
52
52
53
const auto stage1_center_out_dims = trt_common_->getBindingDimensions (2 );
53
54
out_s1center_elem_num_ = std::accumulate (
54
55
stage1_center_out_dims.d + 1 , stage1_center_out_dims.d + stage1_center_out_dims.nbDims , 1 ,
55
56
std::multiplies<int >());
56
57
out_s1center_elem_num_ = out_s1center_elem_num_ * batch_config[2 ];
57
58
out_s1center_elem_num_per_batch_ = static_cast <size_t >(out_s1center_elem_num_ / batch_config[2 ]);
58
- out_s1center_prob_d_ = cuda_utils::make_unique<float []>(out_s1center_elem_num_);
59
+ out_s1center_prob_d_ = autoware:: cuda_utils::make_unique<float []>(out_s1center_elem_num_);
59
60
out_s1center_prob_h_ =
60
- cuda_utils::make_unique_host<float []>(out_s1center_elem_num_, cudaHostAllocPortable);
61
+ autoware:: cuda_utils::make_unique_host<float []>(out_s1center_elem_num_, cudaHostAllocPortable);
61
62
62
63
const auto pred_out_dims = trt_common_->getBindingDimensions (3 );
63
64
out_pred_elem_num_ = std::accumulate (
64
65
pred_out_dims.d + 1 , pred_out_dims.d + pred_out_dims.nbDims , 1 , std::multiplies<int >());
65
66
out_pred_elem_num_ = out_pred_elem_num_ * batch_config[2 ];
66
67
out_pred_elem_num_per_batch_ = static_cast <size_t >(out_pred_elem_num_ / batch_config[2 ]);
67
- out_pred_prob_d_ = cuda_utils::make_unique<float []>(out_pred_elem_num_);
68
+ out_pred_prob_d_ = autoware:: cuda_utils::make_unique<float []>(out_pred_elem_num_);
68
69
out_pred_prob_h_ =
69
- cuda_utils::make_unique_host<float []>(out_pred_elem_num_, cudaHostAllocPortable);
70
+ autoware:: cuda_utils::make_unique_host<float []>(out_pred_elem_num_, cudaHostAllocPortable);
70
71
71
72
g_type_mean_size_ = {{4.6344314 , 1.9600292 , 1.7375569 }, {6.936331 , 2.5178623 , 2.8506238 },
72
73
{11.194943 , 2.9501154 , 3.4918275 }, {12.275775 , 2.9231303 , 3.87086 },
0 commit comments