Searched refs:accum_shift (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1159 int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCell() argument
1325 int left_shift = accum_shift > 0 ? accum_shift : 0; in GEMVForLstmCell()
1326 int right_shift = accum_shift > 0 ? 0 : -accum_shift; in GEMVForLstmCell()
1346 int32 accum_multiplier, int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCellWithSymmetricRange() argument
1371 const int left_shift = accum_shift > 0 ? accum_shift : 0; in GEMVForLstmCellWithSymmetricRange()
1372 const int right_shift = accum_shift > 0 ? 0 : -accum_shift; in GEMVForLstmCellWithSymmetricRange()
3128 int accum_shift = params.accum_shift; in LstmCell() local
3203 bias_data_int32, accum_multiplier, accum_shift, in LstmCell()
3223 scale_stage.result_exponent = accum_shift; in LstmCell()
3404 int32 accum_multiplier, int accum_shift, in LstmCell() argument
[all …]
|
| D | optimized_ops.h | 3073 int accum_shift = params.accum_shift; in LstmCell() local
3161 gemm_params.multiplier_exponent = accum_shift; in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm.cc | 1362 int accum_shift; in Eval() local
1364 &accum_shift); in Eval()
1368 op_params.accum_shift = accum_shift; in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 939 int accum_shift; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 590 int32 accum_multiplier, int accum_shift, in LstmCell() argument
595 op_params.accum_shift = accum_shift; in LstmCell()
|
| D | reference_ops.h | 1344 int accum_shift = params.accum_shift; in LstmCell() local
1431 MultiplyByQuantizedMultiplier(accum, accum_multiplier, accum_shift); in LstmCell()
|