rocAL_pybind.amd.rocal.plugin.tf.ROCAL_iterator Class Reference

• rocAL_pybind
• amd
• rocal
• plugin
• tf
• ROCAL_iterator

Public Member Functions

rocAL_pybind.amd.rocal.plugin.tf.ROCAL_iterator Class Reference

ROCAL iterator for processing images for TF reader. More...

Inheritance diagram for rocAL_pybind.amd.rocal.plugin.tf.ROCAL_iterator:

Public Member Functions
def	__init__ (self, pipelines, size=0, auto_reset=False, fill_last_batch=True, dynamic_shape=False, last_batch_padded=False)
Public Member Functions inherited from rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericImageIterator
def	__init__ (self, pipeline)
def	next (self)
def	__next__ (self)
def	reset (self)
def	__iter__ (self)
def	__del__ (self)

Additional Inherited Members
Data Fields inherited from rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericImageIterator
	loader
	output_list
	bs
	output_tensor_list
	dimensions
	dtype
	output

Detailed Description

ROCAL iterator for processing images for TF reader.

It returns outputs in the form of numpy or cupy arrays.

   @param pipelines            The rocAL pipelines to use for processing data.
   @param size                 The size of the iterator.
   @param auto_reset           Whether to automatically reset the iterator after an epoch.
   @param fill_last_batch      Whether to fill the last batch with repeated data to match the batch size.
   @param dynamic_shape        Whether the iterator supports dynamic shapes.
   @param last_batch_padded    Whether the last batch should be padded to match the batch size.

---

The documentation for this class was generated from the following file:

• /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/docs-6.1.1/rocAL_pybind/amd/rocal/plugin/tf.py