Skip to content
Snippets Groups Projects
Commit 467731fb authored by Hashim Sharif's avatar Hashim Sharif
Browse files

Adding declaration for tensorConvPerfCudaHalf

parent 0926519d
No related branches found
No related tags found
No related merge requests found
...@@ -31,8 +31,14 @@ extern "C"{ ...@@ -31,8 +31,14 @@ extern "C"{
void* tensorConvPerfCuda(void* input, void* filter, void* tensorConvPerfCuda(void* input, void* filter,
int vertical_pad, int horizontal_pad, int vertical_pad, int horizontal_pad,
int vertical_stride, int horizontal_stride, int vertical_stride, int horizontal_stride,
int conv_mode, int conv_groups, int row, int col, int start); int conv_mode, int conv_groups,
int row, int col, int start);
void* tensorConvPerfCudaHalf(void* input_ptr, void* filter_ptr,
int vertical_pad, int horizontal_pad,
int vertical_stride, int horizontal_stride,
int conv_mode, int conv_groups,
int row, int col, int start);
void sampleFilter(Tensor* filter, int skip_rate, int skip_offset); void sampleFilter(Tensor* filter, int skip_rate, int skip_offset);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment