Home
last modified time | relevance | path

Searched refs:output_shift (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h76 const int output_shift = params.output_shift; in FullyConnected() local
105 acc = MultiplyByQuantizedMultiplier(acc, output_multiplier, output_shift); in FullyConnected()
125 const int output_shift = params.output_shift; in FullyConnected() local
158 MultiplyByQuantizedMultiplier(accum, output_multiplier, output_shift); in FullyConnected()
177 const int output_shift = params.output_shift; in ShuffledFullyConnected() local
258 MultiplyByQuantizedMultiplier(acc, output_multiplier, output_shift); in ShuffledFullyConnected()
309 output_shift); in ShuffledFullyConnected()
Dlegacy_reference_ops.h116 int output_shift, int32 output_activation_min, in DepthwiseConv() argument
136 op_params.output_shift = kDepthwiseReverseShift * output_shift; in DepthwiseConv()
150 int output_shift, int32 output_activation_min, in DepthwiseConv() argument
156 output_offset, output_multiplier, output_shift, in DepthwiseConv()
169 int32 output_multiplier, int output_shift, in DepthwiseConv() argument
179 output_offset, output_multiplier, output_shift, in DepthwiseConv()
192 int output_shift, int32 output_activation_min, in DepthwiseConv() argument
198 output_offset, output_multiplier, output_shift, in DepthwiseConv()
280 int32 output_offset, int32 output_multiplier, int output_shift, in Conv() argument
299 op_params.output_shift = kReverseShift * output_shift; in Conv()
[all …]
Ddepthwiseconv_uint8.h112 const int output_shift = params.output_shift; in Run() local
162 output_shift); in Run()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h2298 const int32 output_shift = params.output_shift;
2312 output_shift);
2462 const int32 output_shift = function_params.output_shift;
2476 output_shift);
2603 const int32 output_shift = function_params->output_shift;
2702 acc, output_multiplier, output_shift);
2810 const int32 output_shift = function_params->output_shift;
2893 acc, output_multiplier, output_shift);
2919 acc, output_multiplier, output_shift);
2986 const int32 output_shift = function_params->output_shift;
[all …]
Dlegacy_optimized_ops.h215 int output_shift, int32 output_activation_min, in DepthwiseConv() argument
235 op_params.output_shift = kDepthwiseReverseShift * output_shift; in DepthwiseConv()
249 int output_shift, int32 output_activation_min, in DepthwiseConv() argument
255 output_offset, output_multiplier, output_shift, in DepthwiseConv()
268 int32 output_multiplier, int output_shift, in DepthwiseConv() argument
278 output_offset, output_multiplier, output_shift, in DepthwiseConv()
291 int output_shift, int32 output_activation_min, in DepthwiseConv() argument
297 output_offset, output_multiplier, output_shift, in DepthwiseConv()
361 int output_shift, int32 output_activation_min, in FullyConnected() argument
371 op_params.output_shift = kReverseShift * output_shift; in FullyConnected()
[all …]
Ddepthwiseconv_uint8_3x3_filter.h237 filter_width == 3 && filter_height == 3 && params.output_shift <= 0 &&
407 int32 output_shift;
3477 int32 output_shift) {
3492 (input_depth % 8) == 0 && (output_shift <= 0) &&
3554 const int32 output_shift = rt_params.output_shift;
3570 params.output_right_shift = -output_shift;
4961 const int32 output_shift = function_params->output_shift;
5067 acc0, -output_shift);
5070 acc1, -output_shift);
5073 acc2, -output_shift);
[all …]
Doptimized_ops.h821 int32 output_multiplier, int output_shift, int32 output_activation_min, in FullyConnectedAsGEMVWorkerImpl() argument
832 const bool shift_left = (output_shift > 0); in FullyConnectedAsGEMVWorkerImpl()
1005 const int32 multiplier_power_of_two = 1 << output_shift; in FullyConnectedAsGEMVWorkerImpl()
1013 reduced = RoundingDivideByPOT(reduced, -output_shift); in FullyConnectedAsGEMVWorkerImpl()
1040 int32 output_multiplier, int output_shift, in FullyConnectedAsGEMVWorkerTask()
1055 output_shift_(output_shift), in FullyConnectedAsGEMVWorkerTask()
1097 int32 output_multiplier, int output_shift, int32 output_activation_min, in FullyConnectedAsGEMV() argument
1113 output_shift, output_activation_min, output_activation_max, in FullyConnectedAsGEMV()
1129 output_shift, output_activation_min, output_activation_max, in FullyConnectedAsGEMV()
1177 const int output_shift = params.output_shift; in FullyConnected() local
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h34 const int output_shift = params.output_shift; in FullyConnected() local
57 acc = MultiplyByQuantizedMultiplier(acc, output_multiplier, output_shift); in FullyConnected()
Dconv.h26 const int32* output_shift, const RuntimeShape& input_shape, in ConvPerChannel() argument
113 acc, output_multiplier[out_channel], output_shift[out_channel]); in ConvPerChannel()
Ddepthwise_conv.h24 const int32* output_shift, const RuntimeShape& input_shape, in DepthwiseConvPerChannel() argument
110 output_shift[output_channel]); in DepthwiseConvPerChannel()
Dmul.h34 params.output_shift); in MulElementwise()
116 params.output_shift); in BroadcastMul4DSlow()
Dadd.h51 raw_sum, params.output_multiplier, params.output_shift) + in AddElementwise()
128 raw_sum, params.output_multiplier, params.output_shift) + in BroadcastAdd4DSlow()
/external/tensorflow/tensorflow/core/kernels/
Dquantized_conv_ops.cc60 int output_shift, int output_offset, int output_mult) { in operator ()() argument
71 const int32 rounding = (output_shift < 1) ? 0 : (1 << (output_shift - 1)); in operator ()()
171 output_shift); in operator ()()
206 int output_shift, int output_offset, int output_mult) { in operator ()() argument
226 output_shift, output_offset, output_mult); in operator ()()
389 (output_offset == 0) && (output_mult == 1) && (output_shift == 0) && in operator ()()
396 (output_mult == 1) && (output_shift == 0)) { in operator ()()
442 chunk_output_data, output_shift, output_offset, output_mult, ldc); in operator ()()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dfully_connected.cc34 int output_shift; member
61 data->output_shift = -exponent; in CalculateOpData()
96 op_params.output_shift = -data->output_shift; in EvalQuantized()
Ddepthwise_conv.cc43 int output_shift; member
74 data->output_shift = -exponent; in CalculateOpData()
147 op_params.output_shift = -data->output_shift; in EvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected.cc52 int output_shift; member
126 data->output_shift = -exponent; in Prepare()
301 op_params.output_shift = -data->output_shift; in FullyConnectedInt8()
330 op_params.output_shift = -data->output_shift; \ in EvalQuantized()
409 op_params.output_shift = -data->output_shift; \ in EvalShuffledQuantized()
Dmul.cc50 int output_shift; member
102 real_multiplier, &data->output_multiplier, &data->output_shift); in Prepare()
170 op_params.output_shift = data->output_shift; in EvalQuantized()
Dadd.cc55 int output_shift; member
119 real_output_multiplier, &data->output_multiplier, &data->output_shift); in Prepare()
239 op_params.output_shift = data->output_shift; in EvalAddQuantized()
Ddepthwise_conv.cc57 int output_shift; member
174 &data->output_multiplier, &data->output_shift, in Prepare()
257 op_params.output_shift = -data->output_shift; in EvalQuantized()
Dsub.cc56 int output_shift; member
131 &op_params->output_shift); in Prepare8BitSubOp()
287 op_params.output_shift = data->output_shift; in EvalQuantized()
Dconv.cc71 int output_shift; member
321 &data->output_multiplier, &data->output_shift, in Prepare()
457 op_params.output_shift = -data->output_shift; in EvalQuantized()
484 op_params.output_shift = -data->output_shift; in EvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h370 int* output_shift) { in GetInvSqrtQuantizedMultiplierExp() argument
371 *output_shift = 11; in GetInvSqrtQuantizedMultiplierExp()
374 ++*output_shift; in GetInvSqrtQuantizedMultiplierExp()
381 *output_shift -= left_shift_bit_pairs; in GetInvSqrtQuantizedMultiplierExp()
409 if (*output_shift < 0) { in GetInvSqrtQuantizedMultiplierExp()
410 *output_inv_sqrt <<= -*output_shift; in GetInvSqrtQuantizedMultiplierExp()
411 *output_shift = 0; in GetInvSqrtQuantizedMultiplierExp()
414 *output_shift *= reverse_shift; in GetInvSqrtQuantizedMultiplierExp()
Ddepthwiseconv_quantized_test.cc120 const int output_shift = params.output_shift; in DispatchDepthwiseConv() local
130 pad_height, depth_multiplier, output_shape, output_shift); in DispatchDepthwiseConv()
296 int output_shift, std::int32_t output_activation_min, in TestOneDepthwiseConvWithGivenOutputShift() argument
317 op_params.output_shift = -output_shift; in TestOneDepthwiseConvWithGivenOutputShift()
404 << " output_shift = " << op_params.output_shift; in TestOneDepthwiseConvWithGivenOutputShift()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc46 int output_shift; member
77 data->output_shift = -exponent; in CalculateOpData()
112 const int output_shift = params.output_shift; in DepthwiseConvOptimizedForFilterWidthEight() local
260 acc, output_multiplier, output_shift); in DepthwiseConvOptimizedForFilterWidthEight()
338 op_params.output_shift = -data->output_shift; in EvalQuantized()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h966 int32 output_multiplier, int output_shift, int32 output_activation_min, in FullyConnectedAsGEMV() argument
980 const bool shift_left = (output_shift > 0); in FullyConnectedAsGEMV()
1093 const int32 multiplier_power_of_two = 1 << output_shift; in FullyConnectedAsGEMV()
1101 reduced = RoundingDivideByPOT(reduced, -output_shift); in FullyConnectedAsGEMV()
1160 const int output_shift = params.output_shift; in FullyConnected() local
1180 output_shift, output_activation_min, output_activation_max, in FullyConnected()
1198 bias_data, output_rows, output_offset, output_multiplier, output_shift, in FullyConnected()
1213 int output_shift, int32 output_activation_min, in FullyConnected() argument
1223 op_params.output_shift = kReverseShift * output_shift; in FullyConnected()
1244 const int output_shift = params.output_shift; in FullyConnected() local
[all …]

12