From 536f9eeefc684fd6e6d10d0abbdbe95620189283 Mon Sep 17 00:00:00 2001 From: Valery Chernov Date: Wed, 5 Oct 2022 22:05:24 +0300 Subject: [PATCH] renaming --- onnxruntime/core/providers/tvm/tvm_runner_impl.cc | 4 ++-- onnxruntime/core/providers/tvm/tvm_runner_impl.h | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/onnxruntime/core/providers/tvm/tvm_runner_impl.cc b/onnxruntime/core/providers/tvm/tvm_runner_impl.cc index a1b96f29ead6e..8947e540ed42f 100644 --- a/onnxruntime/core/providers/tvm/tvm_runner_impl.cc +++ b/onnxruntime/core/providers/tvm/tvm_runner_impl.cc @@ -126,7 +126,7 @@ void GERunnerImpl::run() { tvm::TVMRun(*mod_); } -void GERunnerImpl::get_output() { +void GERunnerImpl::get_outputs() { tvm::TVMGetOutputs(*mod_, output_tensors_); } @@ -165,7 +165,7 @@ void VMRunnerImpl::run() { tvm::TVM_VM_Run(*mod_); } -void VMRunnerImpl::get_output() { +void VMRunnerImpl::get_outputs() { tvm::TVM_VM_GetOutputs(*mod_, output_tensors_); } diff --git a/onnxruntime/core/providers/tvm/tvm_runner_impl.h b/onnxruntime/core/providers/tvm/tvm_runner_impl.h index 7ae4a1281bc99..5982bfb127a0a 100644 --- a/onnxruntime/core/providers/tvm/tvm_runner_impl.h +++ b/onnxruntime/core/providers/tvm/tvm_runner_impl.h @@ -55,7 +55,7 @@ class RunnerImpl { set_input(ort, context); connect_output_tensors2ort(ort, context); run(); - get_output(); + get_outputs(); return Status::OK(); } @@ -64,7 +64,7 @@ class RunnerImpl { virtual void connect_output_tensors2ort(Ort::CustomOpApi& ort, OrtKernelContext* context) = 0; virtual void set_output_zero_copy() = 0; virtual void run() = 0; - virtual void get_output() = 0; + virtual void get_outputs() = 0; protected: void convert_input_tensors2dl_tensors(Ort::CustomOpApi& ort, @@ -97,7 +97,7 @@ class GERunnerImpl : public RunnerImpl { void connect_output_tensors2ort(Ort::CustomOpApi& ort, OrtKernelContext* context) final; void set_output_zero_copy() final; void run() final; - void get_output() final; + void get_outputs() final; }; @@ -115,7 +115,7 @@ class VMRunnerImpl : public RunnerImpl { void connect_output_tensors2ort(Ort::CustomOpApi& ort, OrtKernelContext* context) final; void set_output_zero_copy() final; void run() final; - void get_output() final; + void get_outputs() final; private: void infer_once_to_get_output_shapes();