Searched refs:current_cuda_tensor (Results 1 – 1 of 1) sorted by relevance
100 CudaTensorParam ¤t_cuda_tensor = cuda_tensor_map_.find(device_tensor_name)->second; in SyncMemInHostAndDevice() local102 current_cuda_tensor.is_valid_mem = is_host2device ? current_cuda_tensor.is_valid_mem : true; in SyncMemInHostAndDevice()103 if (is_host2device && current_cuda_tensor.is_valid_mem) { in SyncMemInHostAndDevice()107 auto device_ptr = current_cuda_tensor.data; in SyncMemInHostAndDevice()