Lines Matching refs:int32_t
19 inline int32_t MultiplyByQuantizedMultiplier(int32_t x, int32_t quantized_multiplier, int shift) { in MultiplyByQuantizedMultiplier()
30 void MatrixBatchVectorMultiplyAccumulate(const int8_t* input, const int32_t* bias, in MatrixBatchVectorMultiplyAccumulate()
31 const int8_t* input_to_gate_weights, int32_t multiplier, in MatrixBatchVectorMultiplyAccumulate()
32 int32_t shift, int32_t n_batch, int32_t n_input, in MatrixBatchVectorMultiplyAccumulate()
33 int32_t n_output, int32_t output_zp, T* output) { in MatrixBatchVectorMultiplyAccumulate()
38 int32_t acc = bias[row]; in MatrixBatchVectorMultiplyAccumulate()
78 inline bool GetInvSqrtQuantizedMultiplierExp(int32_t input, int reverse_shift, in GetInvSqrtQuantizedMultiplierExp()
79 int32_t* output_inv_sqrt, int* output_shift) { in GetInvSqrtQuantizedMultiplierExp()
87 *output_inv_sqrt = std::numeric_limits<std::int32_t>::max(); in GetInvSqrtQuantizedMultiplierExp()
109 using F3 = FixedPoint<int32_t, 3>; in GetInvSqrtQuantizedMultiplierExp()
110 using F0 = FixedPoint<int32_t, 0>; in GetInvSqrtQuantizedMultiplierExp()
136 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, const int32_t* bias,
137 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, int32_t variance_limit,
140 void MatrixScalarMultiplyAccumulate(const int8_t* matrix, int32_t scalar, int32_t n_row,
141 int32_t n_col, int32_t* output);
143 bool PrecomputeZeroPointTimesWeightWithBias(int32_t zero_point, const int8_t* weight_tensor,
144 const Shape& weight_shape, const int32_t* bias_tensor,
145 std::unique_ptr<int32_t[]>* output);
147 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output);
150 void ApplyTanh(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplyTanh()
163 inline void ApplyTanh(int32_t integer_bits, const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanh()
186 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int32_t multiplier, int32_t shift,
187 int32_t n_batch, int32_t n_input, int32_t output_zp, int8_t* output);
201 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input);
203 void CwiseClipping(int8_t* input, const int8_t clipping_value, int32_t n_batch, int32_t n_input);
207 int32_t multiplier, int shift, int16_t* result);