grid_cpu.h 868 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
void grid_cluster_Float (int C, THLongTensor *output, THFloatTensor  *position,  THFloatTensor *size, THLongTensor *count);
void grid_cluster_Double(int C, THLongTensor *output, THDoubleTensor *position, THDoubleTensor *size, THLongTensor *count);
void grid_cluster_Byte  (int C, THLongTensor *output, THByteTensor   *position,   THByteTensor *size, THLongTensor *count);
void grid_cluster_Char  (int C, THLongTensor *output, THCharTensor   *position,   THCharTensor *size, THLongTensor *count);
void grid_cluster_Short (int C, THLongTensor *output, THShortTensor  *position,  THShortTensor *size, THLongTensor *count);
void grid_cluster_Int   (int C, THLongTensor *output, THIntTensor    *position,    THIntTensor *size, THLongTensor *count);
void grid_cluster_Long  (int C, THLongTensor *output, THLongTensor   *position,   THLongTensor *size, THLongTensor *count);