Lines Matching refs:activation
50 void CalculateActivationRangeImpl(int32_t activation, const Shape& outputShape, int32_t qmin, in CalculateActivationRangeImpl() argument
59 if (activation == kActivationRelu) { in CalculateActivationRangeImpl()
62 } else if (activation == kActivationRelu6) { in CalculateActivationRangeImpl()
65 } else if (activation == kActivationRelu1) { in CalculateActivationRangeImpl()
68 } else if (activation == kActivationNone) { in CalculateActivationRangeImpl()
273 void CalculateActivationRangeUint8(int32_t activation, const Shape& outputShape, int32_t* act_min, in CalculateActivationRangeUint8() argument
278 CalculateActivationRangeImpl(activation, outputShape, qmin, qmax, act_min, act_max); in CalculateActivationRangeUint8()
281 void CalculateActivationRangeInt8(int32_t activation, const Shape& outputShape, int32_t* act_min, in CalculateActivationRangeInt8() argument
286 CalculateActivationRangeImpl(activation, outputShape, qmin, qmax, act_min, act_max); in CalculateActivationRangeInt8()
289 void CalculateActivationRangeFloat(int32_t activation, float* activation_min, in CalculateActivationRangeFloat() argument
291 if (activation == kActivationRelu) { in CalculateActivationRangeFloat()
294 } else if (activation == kActivationRelu6) { in CalculateActivationRangeFloat()
297 } else if (activation == kActivationRelu1) { in CalculateActivationRangeFloat()
300 } else if (activation == kActivationNone) { in CalculateActivationRangeFloat()