Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
chilo-ms committed Dec 5, 2023
1 parent e15a8bd commit 8de13db
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -1108,7 +1108,7 @@ Status BindKernelOutput(Ort::KernelContext& ctx,
break;
}
case ONNX_TENSOR_ELEMENT_DATA_TYPE_DOUBLE: {
// The allocation buffer holds the FLOAT output data since TRT doesn't support DOULBE but FLOAT.
// The allocation buffer holds the FLOAT output data since TRT doesn't support DOUBLE but FLOAT.
// So, we need to cast the data from FLOAT to DOUBEL and then set DOUBLE output data to kernel context.
SafeInt<int> output_dim_size(1);
for (int i = 0; i < shape.size(); ++i) {
Expand Down

0 comments on commit 8de13db

Please sign in to comment.