Searching in torch/utils/benchmark