split_factor

ivy.split_factor(dev_str=None)[source]

Get the global split factor for a given device, which can be used to scale batch splitting chunk sizes for the device across the codebase. Default global value for each device is 1.

Parameters

dev_str (str, optional) – The device to query the split factor for. Sets the default device by default.

Returns

The split factor for the specified device.


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty