We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4959468 commit 9125616Copy full SHA for 9125616
onnxruntime/test/perftest/ort_test_session.h
@@ -37,9 +37,10 @@ class OnnxRuntimeTestSession : public TestSession {
37
Ort::Session session_{nullptr};
38
std::mt19937 rand_engine_;
39
std::uniform_int_distribution<int> dist_;
40
- std::vector<std::vector<Ort::Value>> test_inputs_;
41
OrtAllocator* allocator_ = Ort::AllocatorWithDefaultOptions();
+ // Note: custom_allocator_, if used, must outlive the `Ort::Value`s allocated with it in test_inputs_ and outputs_.
42
Ort::Allocator custom_allocator_{nullptr};
43
+ std::vector<std::vector<Ort::Value>> test_inputs_;
44
std::vector<Ort::Value> outputs_;
45
std::vector<std::string> output_names_;
46
// The same size with output_names_.
0 commit comments