Searched refs:kNeedSyncDeviceToHost (Results 1 – 3 of 3) sorted by relevance
41 enum TensorSyncStatus { kNoNeedSync, kNeedSyncHostToDevice, kNeedSyncDeviceToHost, kNeedSyncDeviceT… enumerator365 bool NeedSyncDeviceToHost() const { return sync_status_ == kNeedSyncDeviceToHost; } in NeedSyncDeviceToHost()
241 tensor->set_sync_status(kNeedSyncDeviceToHost); in CreateNodeOutputTensor()
1622 old_tensors[i]->set_sync_status(kNeedSyncDeviceToHost); in UpdateMsFunctionForwardTensors()