LostTech.TensorFlow : API Documentation

Type Sharding

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

Parent PythonObjectContainer

Interfaces ISharding

Methods

Properties

Public instance methods

void apply_to_tensor(object tensor, bool assign_tuple_sharding)

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

Public static methods

object assign_device_dyn<TClass>(object core)

TClass assign_device<TClass>(int core)

object replicate_dyn<TClass>()

TClass replicate<TClass>()

object split_dyn<TClass>(object tensor, object split_dimension, object num_devices)

TClass split<TClass>(object tensor, object split_dimension, object num_devices)

object tile_dyn<TClass>(object tile_assignment)

TClass tile<TClass>(object tile_assignment)

Public properties

object proto get;

object proto_dyn get;

object PythonObject get;