Complete a device string in configure_collective_ops document.
PiperOrigin-RevId: 275085372 Change-Id: Iff6f255f5a66273a5e5937ee0fce85840378969b
This commit is contained in:
parent
2a8278ebe2
commit
720b4c33ec
@ -606,7 +606,7 @@ class Context(object):
|
||||
|
||||
Args:
|
||||
collective_leader: a device string for collective leader, e.g.
|
||||
"/job:worker/replica:0/task:"; empty string means local execution of
|
||||
"/job:worker/replica:0/task:0"; empty string means local execution of
|
||||
collective ops.
|
||||
scoped_allocator_enabled_ops: a tuple or a list of op names for scoped
|
||||
allocator to run with.
|
||||
|
Loading…
Reference in New Issue
Block a user