Skip to content

Commit

Permalink
renaming
Browse files Browse the repository at this point in the history
  • Loading branch information
Valery Chernov committed Oct 5, 2022
1 parent 7de7c7c commit 536f9ee
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions onnxruntime/core/providers/tvm/tvm_runner_impl.cc
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,7 @@ void GERunnerImpl::run() {
tvm::TVMRun(*mod_);
}

void GERunnerImpl::get_output() {
void GERunnerImpl::get_outputs() {
tvm::TVMGetOutputs(*mod_, output_tensors_);
}

Expand Down Expand Up @@ -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_);
}

Expand Down
8 changes: 4 additions & 4 deletions onnxruntime/core/providers/tvm/tvm_runner_impl.h
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ class RunnerImpl {
set_input(ort, context);
connect_output_tensors2ort(ort, context);
run();
get_output();
get_outputs();

return Status::OK();
}
Expand All @@ -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,
Expand Down Expand Up @@ -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;
};


Expand All @@ -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();
Expand Down

0 comments on commit 536f9ee

Please sign in to comment.