diff options
author | 2018-07-18 18:51:11 -0700 | |
---|---|---|
committer | 2018-07-18 18:54:09 -0700 | |
commit | 6fbfa767b659468b408cacee4aeb2934aa50949e (patch) | |
tree | bcfd0483b69213a737e1bbd32cff8d801c44bfe7 /tensorflow/contrib/lite/kernels/internal | |
parent | e6fb3baeed99543b3ffd22522c30b08c94835e15 (diff) |
add int32 support for div
PiperOrigin-RevId: 205173981
Diffstat (limited to 'tensorflow/contrib/lite/kernels/internal')
-rw-r--r-- | tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h b/tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h index 6fabb9c268..04f61c7434 100644 --- a/tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h +++ b/tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h @@ -1654,10 +1654,11 @@ void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, } } -inline void Div(const float* input1_data, const Dims<4>& input1_dims, - const float* input2_data, const Dims<4>& input2_dims, - float output_activation_min, float output_activation_max, - float* output_data, const Dims<4>& output_dims) { +template <typename T> +inline void Div(const T* input1_data, const Dims<4>& input1_dims, + const T* input2_data, const Dims<4>& input2_dims, + T output_activation_min, T output_activation_max, + T* output_data, const Dims<4>& output_dims) { const int flat_size = MatchingFlatSize(input1_dims, input2_dims, output_dims); for (int i = 0; i < flat_size; ++i) { output_data[i] = ActivationFunctionWithMinMax( |