|
|
|
@@ -398,19 +398,13 @@ void Executor::RunGraphAsync(const SessionPtr &session, const GraphId &graph_id, |
|
|
|
void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info, |
|
|
|
std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs, |
|
|
|
const std::vector<int64_t> &tensors_mask) { |
|
|
|
auto task = std::make_shared<RunOpTask>(); |
|
|
|
task->session_ = session; |
|
|
|
task->op_run_info_ = op_run_info; |
|
|
|
task->graph_info_ = graph_info; |
|
|
|
task->input_tensors_ = input_tensors; |
|
|
|
task->tensors_mask_ = tensors_mask; |
|
|
|
MS_EXCEPTION_IF_NULL(session); |
|
|
|
for (auto &tensor : *input_tensors) { |
|
|
|
if (tensor->NeedWait()) { |
|
|
|
tensor->Wait(); |
|
|
|
} |
|
|
|
} |
|
|
|
RunTask(task, true, true); |
|
|
|
*outputs = task->outputs_; |
|
|
|
session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask); |
|
|
|
} |
|
|
|
|
|
|
|
void Executor::RunOpsInGraph(const SessionPtr &session, const GraphId &graph_id, |
|
|
|
|