|
|
|
@@ -29,6 +29,7 @@ extern "C" { |
|
|
|
void DoStack(const float *const *inputs, size_t input_num, int *in_shape, size_t shape_size, int axis, float *output); |
|
|
|
void DoStackInt32(const int32_t *const *inputs, size_t input_num, int *in_shape, size_t shape_size, int axis, |
|
|
|
int32_t *output); |
|
|
|
void DoStackOneInput(const int8_t *input, int8_t *output, size_t data_size); |
|
|
|
#ifdef __cplusplus |
|
|
|
} |
|
|
|
#endif |
|
|
|
|