(
self,
source=None,
num_outputs=None,
*,
cycle=None,
layout=None,
dtype=None,
ndim=None,
name=None,
device="cpu",
cuda_stream=None,
use_copy_kernel=None,
batch=None,
parallel=None,
no_copy=None,
prefetch_queue_depth=None,
bytes_per_sample_hint=None,
batch_info=None,
repeat_last=False,
**kwargs,
)
| 616 | """ |
| 617 | |
| 618 | def __init__( |
| 619 | self, |
| 620 | source=None, |
| 621 | num_outputs=None, |
| 622 | *, |
| 623 | cycle=None, |
| 624 | layout=None, |
| 625 | dtype=None, |
| 626 | ndim=None, |
| 627 | name=None, |
| 628 | device="cpu", |
| 629 | cuda_stream=None, |
| 630 | use_copy_kernel=None, |
| 631 | batch=None, |
| 632 | parallel=None, |
| 633 | no_copy=None, |
| 634 | prefetch_queue_depth=None, |
| 635 | bytes_per_sample_hint=None, |
| 636 | batch_info=None, |
| 637 | repeat_last=False, |
| 638 | **kwargs, |
| 639 | ): |
| 640 | self._schema = _b.GetSchema("ExternalSource") |
| 641 | self._spec = _b.OpSpec("ExternalSource") |
| 642 | self._device = device |
| 643 | self._layout = layout |
| 644 | self._dtype = dtype |
| 645 | self._ndim = ndim |
| 646 | self._cuda_stream = cuda_stream |
| 647 | self._use_copy_kernel = use_copy_kernel |
| 648 | |
| 649 | import nvidia.dali.ops |
| 650 | |
| 651 | self._init_args, self._call_args = nvidia.dali.ops._separate_kwargs(kwargs) |
| 652 | |
| 653 | if "_module" not in self._init_args: |
| 654 | self._init_args.update({"_module": "nvidia.dali.ops"}) |
| 655 | if "_display_name" not in self._init_args: |
| 656 | self._init_args.update({"_display_name": "ExternalSource"}) |
| 657 | |
| 658 | operator_name = self._operator_name() |
| 659 | |
| 660 | callback, source_desc = _get_callback_from_source(source, cycle, batch_info or False) |
| 661 | |
| 662 | if name is not None and num_outputs is not None: |
| 663 | raise ValueError(f"`num_outputs` is not compatible with named `{operator_name}`") |
| 664 | |
| 665 | self._name = name |
| 666 | self._num_outputs = num_outputs |
| 667 | self._batch = batch |
| 668 | self._callback = callback |
| 669 | self._source_desc = source_desc |
| 670 | self._parallel = parallel |
| 671 | self._no_copy = no_copy |
| 672 | self._prefetch_queue_depth = prefetch_queue_depth |
| 673 | self._bytes_per_sample_hint = bytes_per_sample_hint |
| 674 | self._batch_info = batch_info |
| 675 | self._repeat_last = repeat_last |
nothing calls this directly
no test coverage detected