multi_task for single_op.

pull/1044/head
unknown 4 years ago
parent 84845edb61
commit ac62ccf90e

@ -252,7 +252,7 @@ Status TaskContext::AllocateOutput(int index,
}
if (outputs_start_[index].GetSize() > 0) {
reMemset(outputs_start_[index].MutableData(), outputs_start_[index].GetSize(), 0, outputs_start_[index].GetSize());
rtMemset(outputs_start_[index].MutableData(), outputs_start_[index].GetSize(), 0, outputs_start_[index].GetSize());
}
if (execution_context_->trace_enabled) {
@ -401,7 +401,7 @@ Status TaskContext::PropagateOutputs() {
subgraph_context_->all_inputs_[input_offset] = *tensor;
if (execution_context_->trace_enabled) {
subgraph_context_->all_inputs_[input_offset].SetName(
dst_node_item->NodeName() + "_in_" + std::to_string(dst_input_idx));
dst_node_item->NodeName() + "_in_" + std::to_string(dst_input_idx));
}
}
}

Loading…
Cancel
Save