Merge pull request #23529 from dmatveev:dm/gapi_onnx_rt_1.14.1

Bump supported ONNX RT version to 1.14.1
pull/23563/head
Alexander Smorkalov 2 years ago committed by GitHub
commit 9161e40aa0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 12
      modules/gapi/src/backends/onnx/gonnxbackend.cpp
  2. 10
      modules/gapi/test/infer/gapi_infer_onnx_test.cpp

@ -44,7 +44,8 @@ static std::string pdims(const std::vector<int64_t> &dims) {
struct TensorInfo {
TensorInfo() = default;
explicit TensorInfo(const Ort::TensorTypeAndShapeInfo& info)
explicit TensorInfo(const Ort::ConstTensorTypeAndShapeInfo &info)
: dims(info.GetShape())
, type(info.GetElementType())
, is_dynamic(ade::util::find(dims, -1) != dims.end()) {
@ -688,11 +689,10 @@ std::vector<TensorInfo> ONNXCompiled::getTensorInfo(TensorPosition pos) {
: this_session.GetOutputTypeInfo(i);
tensor_info.emplace_back(info.GetTensorTypeAndShapeInfo());
char *name_p = pos == INPUT
? this_session.GetInputName(i, allocator)
: this_session.GetOutputName(i, allocator);
tensor_info.back().name = name_p;
allocator.Free(name_p);
Ort::AllocatedStringPtr name_p = pos == INPUT
? this_session.GetInputNameAllocated(i, allocator)
: this_session.GetOutputNameAllocated(i, allocator);
tensor_info.back().name = std::string(name_p.get());
}
return tensor_info;

@ -332,9 +332,8 @@ public:
// Inputs Run params
std::vector<Ort::Value> in_tensors;
for(size_t i = 0; i < num_in; ++i) {
char* in_node_name_p = session.GetInputName(i, allocator);
in_node_names.emplace_back(in_node_name_p);
allocator.Free(in_node_name_p);
auto in_node_name_p = session.GetInputNameAllocated(i, allocator);
in_node_names.emplace_back(in_node_name_p.get());
in_node_dims = toORT(ins[i].size);
in_tensors.emplace_back(Ort::Value::CreateTensor<T>(memory_info,
const_cast<T*>(ins[i].ptr<T>()),
@ -345,9 +344,8 @@ public:
// Outputs Run params
if (custom_out_names.empty()) {
for(size_t i = 0; i < num_out; ++i) {
char* out_node_name_p = session.GetOutputName(i, allocator);
out_node_names.emplace_back(out_node_name_p);
allocator.Free(out_node_name_p);
auto out_node_name_p = session.GetOutputNameAllocated(i, allocator);
out_node_names.emplace_back(out_node_name_p.get());
}
} else {
out_node_names = std::move(custom_out_names);

Loading…
Cancel
Save