LostTech.TensorFlow : API Documentation

Type xla_sharding

Namespace tensorflow.compiler.xla.experimental.xla_sharding.xla_sharding

Public static methods

object assign_device(object tensor, int device, bool assign_tuple_sharding)

object assign_device_dyn(object tensor, object device, ImplicitContainer<T> assign_tuple_sharding)

object replicate(object tensor, bool assign_tuple_sharding)

object replicate_dyn(object tensor, ImplicitContainer<T> assign_tuple_sharding)

object split(object tensor, object split_dimension, object num_devices, bool assign_tuple_sharding)

object split_dyn(object tensor, object split_dimension, object num_devices, ImplicitContainer<T> assign_tuple_sharding)

object tile(object tensor, object tile_assignment, bool assign_tuple_sharding)

object tile_dyn(object tensor, object tile_assignment, ImplicitContainer<T> assign_tuple_sharding)

Public properties

PythonFunctionContainer assign_device_fn get;

PythonFunctionContainer replicate_fn get;

PythonFunctionContainer Sharding_fn get;

PythonFunctionContainer split_fn get;