Searched defs:current_cuda_tensor (Results 1 – 1 of 1) sorted by relevance
96 CudaTensorParam ¤t_cuda_tensor = cuda_tensor_map_.find(device_tensor_name)->second; in SyncMemDeviceToHost() local136 CudaTensorParam ¤t_cuda_tensor = it->second; in SyncMemDeviceToHost() local165 CudaTensorParam ¤t_cuda_tensor = cuda_tensor_map_.find(device_tensor_name)->second; in SyncMemInHostAndDevice() local194 CudaTensorParam ¤t_cuda_tensor = cuda_tensor_map_.find(device_tensor_name)->second; in SyncMemInHostAndDevice() local