nni/nas/pytorch/utils.py
nni/retiarii/oneshot/pytorch/utils.py

nni/nas/pytorch/mutables.py
nni/retiarii/nn/pytorch/api.py

nni/nas/tensorflow/mutator.py
nni/retiarii/nn/pytorch/api.py