rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericIteratorDetection Class Reference#
rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericIteratorDetection Class Reference
Iterator for processing data. More...
Inheritance diagram for rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericIteratorDetection:
Detailed Description
Iterator for processing data.
@param pipeline The rocAL pipeline to use for processing data.
@param tensor_layout The layout of the output tensors
@param reverse_channels Whether to reverse the order of color channels.
@param multiplier Multiplier values for color normalization.
@param offset Offset values for color normalization.
@param tensor_dtype Data type of the output tensors.
@param device The device to use for processing
@param device_id The ID of the device to use
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.5/rocAL_pybind/amd/rocal/plugin/tf.py