Commit 40822ddc authored by Robert Kimball's avatar Robert Kimball

create output tensors

parent da7a15f8
...@@ -71,13 +71,17 @@ vector<runtime::PerformanceCounter> run_benchmark_pipelined(shared_ptr<Function> ...@@ -71,13 +71,17 @@ vector<runtime::PerformanceCounter> run_benchmark_pipelined(shared_ptr<Function>
} }
} }
// // Create output tensors for all Results // Create output tensors for all Results
// array<vector<shared_ptr<runtime::Tensor>>, pipeline_depth> output_tensors_array; array<vector<shared_ptr<runtime::Tensor>>, pipeline_depth> output_tensors_array;
// for (shared_ptr<Node> out : f->get_results()) size_t output_index = 0;
// { for (shared_ptr<Node> result : f->get_results())
// auto output_tensors = backend->create_tensor(out->get_element_type(), out->get_shape()); {
// output_tensors_array[i] = output_tensors; auto output_tensors = exec->create_output_tensor(output_index++, pipeline_depth);
// } for(size_t i=0; i<pipeline_depth; i++)
{
output_tensors_array[i].push_back(output_tensors[i]);
}
}
stopwatch t1; stopwatch t1;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment