diff --git a/src/operator/subgraph/tensorrt/nnvm_to_onnx.cc b/src/operator/subgraph/tensorrt/nnvm_to_onnx.cc index 9d98a48c2ec2..3c90d35ecd09 100644 --- a/src/operator/subgraph/tensorrt/nnvm_to_onnx.cc +++ b/src/operator/subgraph/tensorrt/nnvm_to_onnx.cc @@ -519,7 +519,7 @@ void ConvertConstant( auto size = shape.Size(); if (dtype == TensorProto_DataType_FLOAT) { - std::shared_ptr shared_data_ptr(new float[size]); + std::shared_ptr shared_data_ptr(new float[size]); float* const data_ptr = shared_data_ptr.get(); nd.SyncCopyToCPU(static_cast(data_ptr), size); @@ -527,7 +527,7 @@ void ConvertConstant( initializer_proto->add_float_data(data_ptr[blob_idx]); } } else if (dtype == TensorProto_DataType_FLOAT16) { - std::shared_ptr shared_data_ptr(new uint16_t[size]); + std::shared_ptr shared_data_ptr(new uint16_t[size]); uint16_t* const data_ptr = shared_data_ptr.get(); nd.SyncCopyToCPU(static_cast(data_ptr), size); for (size_t blob_idx = 0; blob_idx < size; ++blob_idx) {