LostTech.TensorFlow : API Documentation

Type collective_ops

Namespace tensorflow.python.ops.collective_ops

Methods

Properties

Public static methods

Tensor all_gather(IGraphNodeBase t, int group_size, int group_key, int instance_key, string communication_hint)

object all_gather_dyn(object t, object group_size, object group_key, object instance_key, ImplicitContainer<T> communication_hint)

Tensor all_reduce(IEnumerable<object> t, int group_size, int group_key, int instance_key, string merge_op, string final_op, ImplicitContainer<T> subdiv_offsets, string communication_hint)

Tensor all_reduce(int t, int group_size, int group_key, int instance_key, string merge_op, string final_op, ImplicitContainer<T> subdiv_offsets, string communication_hint)

Tensor all_reduce(IGraphNodeBase t, int group_size, int group_key, int instance_key, string merge_op, string final_op, ImplicitContainer<T> subdiv_offsets, string communication_hint)

Tensor all_reduce(object t, int group_size, int group_key, int instance_key, string merge_op, string final_op, ImplicitContainer<T> subdiv_offsets, string communication_hint)

object all_reduce_dyn(object t, object group_size, object group_key, object instance_key, object merge_op, object final_op, ImplicitContainer<T> subdiv_offsets, ImplicitContainer<T> communication_hint)

Tensor broadcast_recv(TensorShape shape, DType dtype, int group_size, int group_key, int instance_key, string communication_hint)

object broadcast_recv_dyn(object shape, object dtype, object group_size, object group_key, object instance_key, ImplicitContainer<T> communication_hint)

Tensor broadcast_send(IGraphNodeBase t, TensorShape shape, DType dtype, int group_size, int group_key, int instance_key, string communication_hint)

object broadcast_send_dyn(object t, object shape, object dtype, object group_size, object group_key, object instance_key, ImplicitContainer<T> communication_hint)

Public properties

PythonFunctionContainer all_gather_fn get;

PythonFunctionContainer all_reduce_fn get;

PythonFunctionContainer broadcast_recv_fn get;

PythonFunctionContainer broadcast_send_fn get;