Searched refs:set_sync_status (Results 1 – 8 of 8) sorted by relevance
238 cur_loop_tensor->set_sync_status(kNeedSyncHostToDevice); in LoadCtrlInputTensor()257 next_loop_tensor->set_sync_status(kNeedSyncHostToDevice); in LoadCtrlInputTensor()275 epoch_tensor->set_sync_status(kNeedSyncHostToDevice); in LoadCtrlInputTensor()478 tensor->set_sync_status(kNoNeedSync); in LoadInputData()511 tensor->set_sync_status(kNoNeedSync); in LoadInputData()2045 tensor->set_sync_status(kNoNeedSync); in UpdateOutputTensors()2059 tensor->set_sync_status(kNeedSyncHostToDevice); in UpdateOutputTensors()
374 tensor->set_sync_status(kNoNeedSync); in LoadInputData()624 tensor->set_sync_status(kNeedSyncHostToDevice); in UpdateOutputTensors()
232 tensor->set_sync_status(kNoNeedSync); in CreateNodeOutputTensor()239 tensor->set_sync_status(kNeedSyncDeviceToHostImmediately); in CreateNodeOutputTensor()241 tensor->set_sync_status(kNeedSyncDeviceToHost); in CreateNodeOutputTensor()1645 tensor->set_sync_status(kNeedSyncHostToDevice); in UpdateOutputs()1749 tensor->set_sync_status(kNeedSyncHostToDevice); in UpdateOutputTensors()1884 tensor->set_sync_status(kNoNeedSync); in Summary()
359 void set_sync_status(TensorSyncStatus sync_status) { sync_status_ = sync_status; } in set_sync_status() function
238 tensor->set_sync_status(kNeedSyncDeviceToHostImmediately); in CreatTensorForOutput()
270 current_tensor->set_sync_status(kNeedSyncHostToDevice); in RunOpMallocPre()1362 tensor->set_sync_status(kNeedSyncHostToDevice); in SyncNodeOutputTensors()1378 tensor->set_sync_status(kNeedSyncHostToDevice); in SyncNodeOutputTensors()
762 tensor->set_sync_status(kNoNeedSync); in StepLoadCtrlInputs()
1622 old_tensors[i]->set_sync_status(kNeedSyncDeviceToHost); in UpdateMsFunctionForwardTensors()1900 new_tensor->set_sync_status(tensor->sync_status()); in RunOpInVM()