set_split_factor

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

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

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

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


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty