@@ -41,32 +41,32 @@ 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_ = autoware:: cuda_utils::make_unique<float []>(one_hot_input_size * batch_config[2 ]);
51
51
52
52
const auto stage1_center_out_dims = trt_common_->getBindingDimensions (2 );
53
53
out_s1center_elem_num_ = std::accumulate (
54
54
stage1_center_out_dims.d + 1 , stage1_center_out_dims.d + stage1_center_out_dims.nbDims , 1 ,
55
55
std::multiplies<int >());
56
56
out_s1center_elem_num_ = out_s1center_elem_num_ * batch_config[2 ];
57
57
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_);
58
+ out_s1center_prob_d_ = autoware:: cuda_utils::make_unique<float []>(out_s1center_elem_num_);
59
59
out_s1center_prob_h_ =
60
- cuda_utils::make_unique_host<float []>(out_s1center_elem_num_, cudaHostAllocPortable);
60
+ autoware:: cuda_utils::make_unique_host<float []>(out_s1center_elem_num_, cudaHostAllocPortable);
61
61
62
62
const auto pred_out_dims = trt_common_->getBindingDimensions (3 );
63
63
out_pred_elem_num_ = std::accumulate (
64
64
pred_out_dims.d + 1 , pred_out_dims.d + pred_out_dims.nbDims , 1 , std::multiplies<int >());
65
65
out_pred_elem_num_ = out_pred_elem_num_ * batch_config[2 ];
66
66
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_);
67
+ out_pred_prob_d_ = autoware:: cuda_utils::make_unique<float []>(out_pred_elem_num_);
68
68
out_pred_prob_h_ =
69
- cuda_utils::make_unique_host<float []>(out_pred_elem_num_, cudaHostAllocPortable);
69
+ autoware:: cuda_utils::make_unique_host<float []>(out_pred_elem_num_, cudaHostAllocPortable);
70
70
71
71
g_type_mean_size_ = {{4.6344314 , 1.9600292 , 1.7375569 }, {6.936331 , 2.5178623 , 2.8506238 },
72
72
{11.194943 , 2.9501154 , 3.4918275 }, {12.275775 , 2.9231303 , 3.87086 },
0 commit comments