saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-03 03:58:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[VitisAI] fix tensor has multi data type (#18188)

Browse source 
### Description
<!-- Describe your changes. -->
When take a tensor's data as raw, clear data with other types within the
tensor.


### Motivation and Context
<!-- - Why is this change required? What problem does it solve? -->
One model's graph transformation caused a node with multiple data types.
This would make the model valid.
This commit is contained in:
BoarQing 2023-11-06 23:16:17 +08:00 committed by GitHub
parent dfafcb58aa
commit d652b1fe48
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/core/providers/vitisai/imp/tensor_proto.cc
View file

@ -16,6 +16,12 @@ gsl::span<const char> tensor_proto_as_raw(
std::vector<uint8_t> unpacked_tensor;
auto s = onnxruntime::utils::UnpackInitializerData(tensor, onnxruntime::Path(), unpacked_tensor);
mut_tensor.mutable_raw_data()->resize(unpacked_tensor.size());
mut_tensor.clear_float_data();
mut_tensor.clear_int32_data();
mut_tensor.clear_string_data();
mut_tensor.clear_int64_data();
mut_tensor.clear_double_data();
mut_tensor.clear_uint64_data();
memcpy(mut_tensor.mutable_raw_data()->data(), unpacked_tensor.data(), unpacked_tensor.size());
}
return gsl::span<const char>(tensor.raw_data().data(), tensor.raw_data().size());