|
47 | 47 | /** */ \
|
48 | 48 | /** Function "__name__" executes operator "__operation1__" over corresponding elements of input arrays */ \
|
49 | 49 | /** */ \
|
50 |
| - /** @param[in] array1 Input array 1. */ \ |
51 |
| - /** @param[in] array2 Input array 2. */ \ |
52 |
| - /** @param[out] result1 Output array. */ \ |
53 |
| - /** @param[in] size Number of elements in the output array. */ \ |
| 50 | + /** @param[out] result_out Output array. */ \ |
| 51 | + /** @param[in] result_size Output array size. */ \ |
| 52 | + /** @param[in] result_ndim Number of output array dimensions. */ \ |
| 53 | + /** @param[in] result_shape Output array shape. */ \ |
| 54 | + /** @param[in] result_strides Output array strides. */ \ |
| 55 | + /** @param[in] input1_in Input array 1. */ \ |
| 56 | + /** @param[in] input1_size Input array 1 size. */ \ |
| 57 | + /** @param[in] input1_ndim Number of input array 1 dimensions. */ \ |
| 58 | + /** @param[in] input1_shape Input array 1 shape. */ \ |
| 59 | + /** @param[in] input1_strides Input array 1 strides. */ \ |
| 60 | + /** @param[in] input2_in Input array 2. */ \ |
| 61 | + /** @param[in] input2_size Input array 2 size. */ \ |
| 62 | + /** @param[in] input2_ndim Number of input array 2 dimensions. */ \ |
| 63 | + /** @param[in] input2_shape Input array 2 shape. */ \ |
| 64 | + /** @param[in] input2_strides Input array 2 strides. */ \ |
| 65 | + /** @param[in] where Where condition. */ \ |
54 | 66 | template <typename _DataType_input1, typename _DataType_input2, typename _DataType_output> \
|
55 |
| - void __name__(void* array1, void* array2, void* result1, size_t size); |
| 67 | + void __name__(void* result_out, \ |
| 68 | + const size_t result_size, \ |
| 69 | + const size_t result_ndim, \ |
| 70 | + const size_t* result_shape, \ |
| 71 | + const size_t* result_strides, \ |
| 72 | + const void* input1_in, \ |
| 73 | + const size_t input1_size, \ |
| 74 | + const size_t input1_ndim, \ |
| 75 | + const size_t* input1_shape, \ |
| 76 | + const size_t* input1_strides, \ |
| 77 | + const void* input2_in, \ |
| 78 | + const size_t input2_size, \ |
| 79 | + const size_t input2_ndim, \ |
| 80 | + const size_t* input2_shape, \ |
| 81 | + const size_t* input2_strides, \ |
| 82 | + const size_t* where) |
56 | 83 |
|
57 | 84 | #endif
|
58 | 85 |
|
|
0 commit comments