LostTech.TensorFlow : API Documentation

Type gen_nccl_ops

Namespace tensorflow.python.ops.gen_nccl_ops

Public static methods

Tensor nccl_all_reduce(IGraphNodeBase input, string reduction, int num_devices, string shared_name, string name)

object nccl_all_reduce_dyn(object input, object reduction, object num_devices, object shared_name, object name)

object nccl_all_reduce_eager_fallback(IGraphNodeBase input, Byte[] reduction, int num_devices, Byte[] shared_name, string name, Context ctx)

object nccl_all_reduce_eager_fallback(IGraphNodeBase input, Byte[] reduction, int num_devices, string shared_name, string name, Context ctx)

object nccl_all_reduce_eager_fallback(IGraphNodeBase input, string reduction, int num_devices, Byte[] shared_name, string name, Context ctx)

object nccl_all_reduce_eager_fallback(IGraphNodeBase input, string reduction, int num_devices, string shared_name, string name, Context ctx)

object nccl_all_reduce_eager_fallback_dyn(object input, object reduction, object num_devices, object shared_name, object name, object ctx)

Tensor nccl_broadcast(IGraphNodeBase input, TensorShape shape, string name)

object nccl_broadcast_dyn(object input, object shape, object name)

object nccl_broadcast_eager_fallback(IGraphNodeBase input, IEnumerable<Nullable<int>> shape, string name, Context ctx)

object nccl_broadcast_eager_fallback(IGraphNodeBase input, TensorShape shape, string name, Context ctx)

object nccl_broadcast_eager_fallback_dyn(object input, object shape, object name, object ctx)

Tensor nccl_reduce(IEnumerable<object> input, string reduction, string name)

object nccl_reduce_dyn(object input, object reduction, object name)

object nccl_reduce_eager_fallback(IEnumerable<object> input, Byte[] reduction, string name, Context ctx)

object nccl_reduce_eager_fallback(IEnumerable<object> input, string reduction, string name, Context ctx)

object nccl_reduce_eager_fallback_dyn(object input, object reduction, object name, object ctx)

Public properties

PythonFunctionContainer nccl_all_reduce_eager_fallback_fn get;

PythonFunctionContainer nccl_all_reduce_fn get;

PythonFunctionContainer nccl_broadcast_eager_fallback_fn get;

PythonFunctionContainer nccl_broadcast_fn get;

PythonFunctionContainer nccl_reduce_eager_fallback_fn get;

PythonFunctionContainer nccl_reduce_fn get;