|
|
@@ -162,6 +162,7 @@ tensor::TensorPtr CPUKernelRuntime::CreatTensorForOutput(session::KernelGraph *k |
|
|
tensor->set_device_address(address); |
|
|
tensor->set_device_address(address); |
|
|
need_sync_outputs->emplace_back(tensor); |
|
|
need_sync_outputs->emplace_back(tensor); |
|
|
} else { |
|
|
} else { |
|
|
|
|
|
tensor->set_device_address(nullptr); |
|
|
address->ptr_ = tensor->data_c(); |
|
|
address->ptr_ = tensor->data_c(); |
|
|
} |
|
|
} |
|
|
address->ref_count_ = INIT_NODE_REF; |
|
|
address->ref_count_ = INIT_NODE_REF; |
|
|
|