diff options
Diffstat (limited to 'compiler/luci-interpreter/pal/cmsisnn/PALMul.h')
-rw-r--r-- | compiler/luci-interpreter/pal/cmsisnn/PALMul.h | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/compiler/luci-interpreter/pal/cmsisnn/PALMul.h b/compiler/luci-interpreter/pal/cmsisnn/PALMul.h index 2b46b100c..347a97a83 100644 --- a/compiler/luci-interpreter/pal/cmsisnn/PALMul.h +++ b/compiler/luci-interpreter/pal/cmsisnn/PALMul.h @@ -21,21 +21,21 @@ namespace luci_interpreter_pal { +template <typename T> static inline void Mul(tflite::ArithmeticParams ¶ms, const tflite::RuntimeShape &input1_shape, - const float *input1_data, const tflite::RuntimeShape &input2_shape, - const float *input2_data, const tflite::RuntimeShape &output_shape, - float *output_data) + const T *input1_data, const tflite::RuntimeShape &input2_shape, + const T *input2_data, const tflite::RuntimeShape &output_shape, + T *output_data) { tflite::reference_ops::BroadcastMul4DSlow(params, input1_shape, input1_data, input2_shape, input2_data, output_shape, output_data); } -static inline void BroadcastMul4DSlow(tflite::ArithmeticParams ¶ms, - const tflite::RuntimeShape &input1_shape, - const float *input1_data, - const tflite::RuntimeShape &input2_shape, - const float *input2_data, - const tflite::RuntimeShape &output_shape, float *output_data) +template <typename T> +static inline void +BroadcastMul4DSlow(tflite::ArithmeticParams ¶ms, const tflite::RuntimeShape &input1_shape, + const T *input1_data, const tflite::RuntimeShape &input2_shape, + const T *input2_data, const tflite::RuntimeShape &output_shape, T *output_data) { tflite::reference_ops::BroadcastMul4DSlow(params, input1_shape, input1_data, input2_shape, input2_data, output_shape, output_data); |