Searched refs:weights_zero_point (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 418 const uint8* weights_data, uint8 weights_zero_point, in GEMVForLstmCell() argument
448 const int16x8_t weights_offset_vec = vdupq_n_s16(-weights_zero_point); in GEMVForLstmCell()
3809 int32 weights_zero_point = params.weights_zero_point; in LstmCell() local
3885 weights_data_uint8, weights_zero_point, bias_shape, in LstmCell()
3913 -weights_zero_point, -128, output_pipeline); in LstmCell()
4088 const Dims<4>& activ_temp_dims, int32 weights_zero_point, in LstmCell() argument
4092 op_params.weights_zero_point = weights_zero_point; in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 853 const Dims<4>& activ_temp_dims, int32 weights_zero_point, in LstmCell() argument
857 op_params.weights_zero_point = weights_zero_point; in LstmCell()
|
| D | optimized_ops.h | 296 const uint8* weights_data, uint8 weights_zero_point, in GEMVForLstmCell() argument
326 const int16x8_t weights_offset_vec = vdupq_n_s16(-weights_zero_point); in GEMVForLstmCell()
3693 int32 weights_zero_point = params.weights_zero_point; in LstmCell() local
3769 weights_data_uint8, weights_zero_point, bias_shape, in LstmCell()
3797 -weights_zero_point, -128, output_pipeline); in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 856 int32 weights_zero_point; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 879 int32 weights_zero_point; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 552 const Dims<4>& activ_temp_dims, int32 weights_zero_point, in LstmCell() argument
556 op_params.weights_zero_point = weights_zero_point; in LstmCell()
|
| D | reference_ops.h | 2055 int32 weights_zero_point = params.weights_zero_point; in LstmCell() local
2136 weights_data_uint8[out_c * fc_accum_depth + d] - weights_zero_point; in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm.cc | 799 op_params.weights_zero_point = weights->params.zero_point; in Eval()
|