nni/algorithms/nas/pytorch/enas/mutator.py
nni/retiarii/oneshot/pytorch/enas.py

nni/algorithms/nas/pytorch/darts/trainer.py
nni/retiarii/oneshot/pytorch/darts.py

nni/algorithms/nas/pytorch/proxylessnas/mutator.py
nni/retiarii/oneshot/pytorch/proxyless.py

nni/algorithms/nas/pytorch/spos/trainer.py
nni/retiarii/oneshot/pytorch/random.py

nni/algorithms/nas/pytorch/enas/trainer.py
nni/retiarii/oneshot/pytorch/enas.py

nni/algorithms/compression/pytorch/auto_compress/utils.py
nni/retiarii/utils.py