Searched refs:output_scale_log2_rounded (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add.cc | 153 int output_scale_log2_rounded; in Prepare() local
155 CheckedLog2(output->params.scale, &output_scale_log2_rounded); in Prepare()
158 data->input1_shift = input1_scale_log2_rounded - output_scale_log2_rounded; in Prepare()
159 data->input2_shift = input2_scale_log2_rounded - output_scale_log2_rounded; in Prepare()
|
| D | sub.cc | 170 int output_scale_log2_rounded; in PrepareInt16SubOp() local
172 CheckedLog2(output->params.scale, &output_scale_log2_rounded); in PrepareInt16SubOp()
175 data->input1_shift = input1_scale_log2_rounded - output_scale_log2_rounded; in PrepareInt16SubOp()
176 data->input2_shift = input2_scale_log2_rounded - output_scale_log2_rounded; in PrepareInt16SubOp()
|
| D | activations.cc | 156 int output_scale_log2_rounded; in TanhPrepare() local
158 context, CheckedLog2(output->params.scale, &output_scale_log2_rounded)); in TanhPrepare()
159 TF_LITE_ENSURE_EQ(context, output_scale_log2_rounded, in TanhPrepare()
219 int output_scale_log2_rounded; in SigmoidPrepare() local
221 context, CheckedLog2(output->params.scale, &output_scale_log2_rounded)); in SigmoidPrepare()
222 TF_LITE_ENSURE_EQ(context, output_scale_log2_rounded, in SigmoidPrepare()
|