| 373 | } |
| 374 | |
| 375 | daliResult_t daliTensorListGetStream( |
| 376 | daliTensorList_h tensor_list, |
| 377 | cudaStream_t *out_stream) { |
| 378 | DALI_PROLOG(); |
| 379 | auto *ptr = ToPointer(tensor_list); |
| 380 | CHECK_OUTPUT(out_stream); |
| 381 | auto str = ptr->GetStream(); |
| 382 | *out_stream = str.has_value() ? *str : cudaStream_t(-1); |
| 383 | return str.has_value() ? DALI_SUCCESS : DALI_NO_DATA; |
| 384 | DALI_EPILOG(); |
| 385 | } |
| 386 | |
| 387 | daliResult_t daliTensorListSetStream( |
| 388 | daliTensorList_h tensor_list, |