fps_cpu.h 160 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
#pragma once

#include <torch/extension.h>

rusty1s's avatar
rusty1s committed
5
torch::Tensor fps_cpu(torch::Tensor src, torch::Tensor ptr, double ratio,
rusty1s's avatar
rusty1s committed
6
                      bool random_start);