Searched refs:TensorParallelRNGTracker (Results 1 – 2 of 2) sorted by relevance
11 TensorParallelRNGTracker,74 random._rng_tracker, TensorParallelRNGTracker76 random._rng_tracker = TensorParallelRNGTracker(device_mesh.device_type)
92 if isinstance(_rng_tracker, TensorParallelRNGTracker):342 class TensorParallelRNGTracker(_RNGStateTracker): class