set_split_factor

ivy.set_split_factor(factor, dev_str=None)[source]

Set the global split factor for a given device, which can be used to scale batch splitting chunk sizes for the device across the codebase.

Parameters
  • factor (float) – The factor to set the device-specific split factor to.

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


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty