Home
last modified time | relevance | path

Searched refs:TensorParallelRNGTracker (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/
H A Dapi.py11 TensorParallelRNGTracker,
74 random._rng_tracker, TensorParallelRNGTracker
76 random._rng_tracker = TensorParallelRNGTracker(device_mesh.device_type)
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_random.py92 if isinstance(_rng_tracker, TensorParallelRNGTracker):
342 class TensorParallelRNGTracker(_RNGStateTracker): class