|
122 | 122 | #define INCLUDE_ONNXRUNTIME_CORE_PROVIDERS_UTILS_ORT_GRAPH_TO_PROTO_H_ |
123 | 123 |
|
124 | 124 | #include <functional> |
125 | | -#include "onnxruntime_cxx_api.h" |
| 125 | +#include "core/session/onnxruntime_cxx_api.h" |
126 | 126 | #include "onnx/onnx_pb.h" |
127 | 127 |
|
128 | 128 | namespace OrtEpUtils { |
@@ -232,7 +232,7 @@ static Ort::Status GetOrtValueInfoTensorTypeShape(const OrtValueInfo& ort_value_ |
232 | 232 | /*out*/ std::vector<int64_t>& dims, |
233 | 233 | /*out*/ std::vector<std::string>& symbolic_dims); |
234 | 234 | static Ort::Status OrtValueInfoToProto(const OrtValueInfo& ort_value_info, onnx::ValueInfoProto& value_info_proto); |
235 | | -static Ort::Status OrtOpAttrToProto(const OrtNode& ort_node, const OrtOpAttr& ort_attr, onnx::AttributeProto& attr_proto); |
| 235 | +static Ort::Status OrtOpAttrToProto(const OrtOpAttr& ort_attr, onnx::AttributeProto& attr_proto); |
236 | 236 |
|
237 | 237 | Ort::Status OrtGraphToProto(const OrtGraph& ort_graph, |
238 | 238 | onnx::GraphProto& graph_proto, |
@@ -379,7 +379,7 @@ Ort::Status OrtGraphToProto(const OrtGraph& ort_graph, |
379 | 379 | } |
380 | 380 |
|
381 | 381 | onnx::AttributeProto* attr_proto = node_proto->add_attribute(); |
382 | | - ORT_EP_UTILS_CXX_RETURN_IF_ERROR(OrtOpAttrToProto(*ort_node, *ort_attr, *attr_proto)); |
| 382 | + ORT_EP_UTILS_CXX_RETURN_IF_ERROR(OrtOpAttrToProto(*ort_attr, *attr_proto)); |
383 | 383 | } |
384 | 384 | } |
385 | 385 |
|
@@ -652,7 +652,7 @@ static Ort::Status OrtValueInfoToProto(const OrtValueInfo& ort_value_info, |
652 | 652 | return Ort::Status{nullptr}; |
653 | 653 | } |
654 | 654 |
|
655 | | -static Ort::Status OrtOpAttrToProto(const OrtNode& ort_node, const OrtOpAttr& ort_attr, onnx::AttributeProto& attr_proto) { |
| 655 | +static Ort::Status OrtOpAttrToProto(const OrtOpAttr& ort_attr, onnx::AttributeProto& attr_proto) { |
656 | 656 | const OrtApi& ort_api = Ort::GetApi(); |
657 | 657 |
|
658 | 658 | const char* attr_name = nullptr; |
@@ -766,7 +766,7 @@ static Ort::Status OrtOpAttrToProto(const OrtNode& ort_node, const OrtOpAttr& or |
766 | 766 | // TensorProto as an attribute value doesn't require a name. |
767 | 767 |
|
768 | 768 | OrtValue* ort_value = nullptr; |
769 | | - ORT_EP_UTILS_C_RETURN_IF_ERROR(ort_api.Node_GetTensorAttributeAsOrtValue(&ort_node, &ort_attr, &ort_value)); |
| 769 | + ORT_EP_UTILS_C_RETURN_IF_ERROR(ort_api.OpAttr_GetTensorAttributeAsOrtValue(&ort_attr, &ort_value)); |
770 | 770 |
|
771 | 771 | Ort::Value tensor(ort_value); |
772 | 772 |
|
|
0 commit comments