Skip to content
Snippets Groups Projects
Commit 8d8857d7 authored by hashimsharif's avatar hashimsharif
Browse files

Adding CPU tensor operations and approximations

parent 8febde45
No related branches found
No related tags found
No related merge requests found
......@@ -22,17 +22,18 @@ extern "C"{
// NOTE: Currently only using 4-D tensors - 2D and 3D tensors not supported for cuDNN operations
// NOTE: The only data format supported as of now is: NCHW (batch_dimension, channels, Height, Width)
void* create4DTensor(int data_type, int data_format, size_t dim1_size, size_t dim2_size,
size_t dim3_size, size_t dim4_size);
size_t dim3_size, size_t dim4_size, bool freeMemory = true);
void initTensorData(void* tensor, void* data_ptr, size_t size_in_bytes);
/********** Tensor Operation API ******/
// NOTE: For conv_mode, only value '1' is supported
void* tensorConvolutionCPU(void* input, void* filter,
int vertical_pad, int horizontal_pad,
int vertical_stride, int horizontal_stride,
int conv_mode, int compute_precision);
void* tensorConvolutionCPU(void *input_ptr, void *filter_ptr,
int vertical_pad, int horizontal_pad,
int vertical_stride, int horizontal_stride,
int conv_mode, int compute_precision,
int row, int col, int skip_every, int start);
void* tensorPoolingCPU(void* input,
int poolFunction,
......
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