From 8edec47c6c84a0baa17891c0ba659151e06beb99 Mon Sep 17 00:00:00 2001 From: mingyueliuh <131847423+mingyueliuh@users.noreply.github.com> Date: Fri, 14 Jun 2024 06:08:51 +0800 Subject: [PATCH] [VitisAI] Fix some typos in tensor_proto_new APIs (#21027) ### Description Fix some typos in Vitis AI EP. ### Motivation and Context Co-authored-by: liumingyue --- onnxruntime/core/providers/vitisai/imp/tensor_proto.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/vitisai/imp/tensor_proto.cc b/onnxruntime/core/providers/vitisai/imp/tensor_proto.cc index 63aa1daf7e..f53894b9d1 100644 --- a/onnxruntime/core/providers/vitisai/imp/tensor_proto.cc +++ b/onnxruntime/core/providers/vitisai/imp/tensor_proto.cc @@ -68,7 +68,7 @@ ONNX_NAMESPACE::TensorProto* tensor_proto_new_i32(const std::string& name, const } ONNX_NAMESPACE::TensorProto* tensor_proto_new_i64(const std::string& name, const std::vector& shape, const std::vector& data) { - return tensor_proto_new(name, shape, ONNX_NAMESPACE::TensorProto_DataType_INT32, + return tensor_proto_new(name, shape, ONNX_NAMESPACE::TensorProto_DataType_INT64, reinterpret_cast(&data[0]), data.size() * sizeof(data[0])); } ONNX_NAMESPACE::TensorProto* tensor_proto_new_u8(const std::string& name, const std::vector& shape, @@ -88,7 +88,7 @@ ONNX_NAMESPACE::TensorProto* tensor_proto_new_u32(const std::string& name, const } ONNX_NAMESPACE::TensorProto* tensor_proto_new_u64(const std::string& name, const std::vector& shape, const std::vector& data) { - return tensor_proto_new(name, shape, ONNX_NAMESPACE::TensorProto_DataType_UINT32, + return tensor_proto_new(name, shape, ONNX_NAMESPACE::TensorProto_DataType_UINT64, reinterpret_cast(&data[0]), data.size() * sizeof(data[0])); }