You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
12 lines
198 B
C
12 lines
198 B
C
4 years ago
|
#ifndef _SEARCHSORTED_CPU
|
||
|
#define _SEARCHSORTED_CPU
|
||
|
|
||
|
#include <torch/extension.h>
|
||
|
|
||
|
void searchsorted_cpu_wrapper(
|
||
|
at::Tensor a,
|
||
|
at::Tensor v,
|
||
|
at::Tensor res,
|
||
|
bool side_left);
|
||
|
|
||
|
#endif
|