diff --git a/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/approx_api.h b/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/approx_api.h index 5bae5b0f4cf55e7013e2994b1469ffed295367c9..ed1c373f6557f08897f6fd16ae7e1b5d17c3c28e 100644 --- a/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/approx_api.h +++ b/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/approx_api.h @@ -27,8 +27,10 @@ extern "C"{ int conv_mode, int conv_groups, int row, int col); void* tensorConvolutionKernelSamp(void* input, void* filter_ptr, - int vertical_pad, int horizontal_pad, int vertical_stride, - int horizontal_stride, int conv_mode, int conv_groups, int skip_every); + int vertical_pad, int horizontal_pad, + int vertical_stride, int horizontal_stride, + int conv_mode, int conv_groups, + int skip_every); void* tensorConvPerfCuda(void* input, void* filter, int vertical_pad, int horizontal_pad, @@ -50,17 +52,23 @@ extern "C"{ int conv_mode, int conv_groups, int skip_rate, int skip_offset); - void *autotuner_tensorFft(void *input, bool inverse); + + + void *autotuner_tensorFft(void *input, bool inverse); + - void *autotuner_tensorReduce(void *input, size_t axis, MathOp func); + void *autotuner_tensorReduce(void *input, size_t axis, MathOp func); - void *autotuner_tensorProjectiveT(void *input, void *transformation); - void *autotuner_tensorMap1(MathOp func, void *input); + void *autotuner_tensorProjectiveT(void *input, void *transformation); + + + void *autotuner_tensorMap1(MathOp func, void *input); - void *autotuner_tensorMap2(MathOp func, void *input1, void *input2); + void *autotuner_tensorMap2(MathOp func, void *input1, void *input2); - void *autotuner_tensorMap3(MathOp func, void *input1, void *input2, void *input3); + + void *autotuner_tensorMap3(MathOp func, void *input1, void *input2, void *input3); void* tensorConvInputHalf(void* input_ptr, void* filter_ptr, int vertical_pad, int horizontal_pad, int vertical_stride,