Spaces:
Build error
Build error
File size: 865 Bytes
8f87579 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
/*
* File : prroi_pooling_gpu.h
* Author : Jiayuan Mao, Tete Xiao
* Email : maojiayuan@gmail.com, jasonhsiao97@gmail.com
* Date : 07/13/2018
*
* Distributed under terms of the MIT license.
* Copyright (c) 2017 Megvii Technology Limited.
*/
int prroi_pooling_forward_cuda(THCudaTensor *features, THCudaTensor *rois, THCudaTensor *output, int pooled_height, int pooled_width, float spatial_scale);
int prroi_pooling_backward_cuda(
THCudaTensor *features, THCudaTensor *rois, THCudaTensor *output, THCudaTensor *output_diff, THCudaTensor *features_diff,
int pooled_height, int pooled_width, float spatial_scale
);
int prroi_pooling_coor_backward_cuda(
THCudaTensor *features, THCudaTensor *rois, THCudaTensor *output, THCudaTensor *output_diff, THCudaTensor *features_diff,
int pooled_height, int pooled_width, float spatial_scal
);
|