Searching in torch/jit