rocAL_pybind.amd.rocal.plugin.pytorch.ROCALNumpyIterator Class Reference#
rocAL_pybind.amd.rocal.plugin.pytorch.ROCALNumpyIterator Class Reference
Inheritance diagram for rocAL_pybind.amd.rocal.plugin.pytorch.ROCALNumpyIterator:
Public Member Functions | |
| def | __init__ (self, pipeline, device="cpu", device_id=0) |
| def | next (self) |
| def | __next__ (self) |
| def | reset (self) |
| def | __iter__ (self) |
| def | __len__ (self) |
| def | __del__ (self) |
Data Fields | |
| loader | |
| device | |
| device_id | |
| output_memory_type | |
| output_list | |
| batch_size | |
| len | |
| last_batch_policy | |
| last_batch_size | |
| output_tensor_list | |
The documentation for this class was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/latest/rocAL_pybind/amd/rocal/plugin/pytorch.py