Searched refs:accum_shift (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm.cc | 795 int accum_shift; in Eval() local
797 &accum_shift); in Eval()
801 op_params.accum_shift = accum_shift; in Eval()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 421 int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCell() argument
587 int left_shift = accum_shift > 0 ? accum_shift : 0; in GEMVForLstmCell()
588 int right_shift = accum_shift > 0 ? 0 : -accum_shift; in GEMVForLstmCell()
608 int32 accum_multiplier, int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCellWithSymmetricRange() argument
633 const int left_shift = accum_shift > 0 ? accum_shift : 0; in GEMVForLstmCellWithSymmetricRange()
634 const int right_shift = accum_shift > 0 ? 0 : -accum_shift; in GEMVForLstmCellWithSymmetricRange()
3811 int accum_shift = params.accum_shift; in LstmCell() local
3886 bias_data_int32, accum_multiplier, accum_shift, in LstmCell()
3906 scale_stage.result_exponent = accum_shift; in LstmCell()
4089 int32 accum_multiplier, int accum_shift, in LstmCell() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 299 int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCell() argument
465 int left_shift = accum_shift > 0 ? accum_shift : 0; in GEMVForLstmCell()
466 int right_shift = accum_shift > 0 ? 0 : -accum_shift; in GEMVForLstmCell()
486 int32 accum_multiplier, int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCellWithSymmetricRange() argument
511 const int left_shift = accum_shift > 0 ? accum_shift : 0; in GEMVForLstmCellWithSymmetricRange()
512 const int right_shift = accum_shift > 0 ? 0 : -accum_shift; in GEMVForLstmCellWithSymmetricRange()
3695 int accum_shift = params.accum_shift; in LstmCell() local
3770 bias_data_int32, accum_multiplier, accum_shift, in LstmCell()
3790 scale_stage.result_exponent = accum_shift; in LstmCell()
|
| D | legacy_optimized_ops.h | 854 int32 accum_multiplier, int accum_shift, in LstmCell() argument
859 op_params.accum_shift = accum_shift; in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 858 int accum_shift; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 881 int accum_shift; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 553 int32 accum_multiplier, int accum_shift, in LstmCell() argument
558 op_params.accum_shift = accum_shift; in LstmCell()
|
| D | reference_ops.h | 2057 int accum_shift = params.accum_shift; in LstmCell() local
2144 MultiplyByQuantizedMultiplier(accum, accum_multiplier, accum_shift); in LstmCell()
|