Divide the input batch into [n_gpus] slices, and obtain slice no. [part]. i.e. if len(x)=10, then slice_batch(x, 2, 1) will return x[5:].
(x, n_gpus, part)
source not stored for this graph (policy: none)
nothing calls this directly
no outgoing calls
no test coverage detected