diff options
author | 2018-06-26 23:41:11 -0700 | |
---|---|---|
committer | 2018-06-28 21:37:43 -0700 | |
commit | c30c57bd0792c50397883252ee5b2960988846d3 (patch) | |
tree | 5412a82ddd4ef7e3dde8c8389c88321e49608561 /tensorflow/contrib/lite/kernels/div.cc | |
parent | 92cc6352abaf2442c0d29755f87f6dbcd514a684 (diff) |
add int32 support for add
PiperOrigin-RevId: 202259189
Diffstat (limited to 'tensorflow/contrib/lite/kernels/div.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/div.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/div.cc b/tensorflow/contrib/lite/kernels/div.cc index d264821e30..bc5c3783fd 100644 --- a/tensorflow/contrib/lite/kernels/div.cc +++ b/tensorflow/contrib/lite/kernels/div.cc @@ -83,8 +83,8 @@ void EvalFloat(TfLiteContext* context, TfLiteNode* node, const TfLiteTensor* input1, const TfLiteTensor* input2, TfLiteTensor* output) { float output_activation_min, output_activation_max; - CalculateActivationRangeFloat(params->activation, &output_activation_min, - &output_activation_max); + CalculateActivationRange(params->activation, &output_activation_min, + &output_activation_max); #define TF_LITE_DIV(type, opname) \ type::opname(GetTensorData<float>(input1), GetTensorDims(input1), \ GetTensorData<float>(input2), GetTensorDims(input2), \ |