MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / CalculateActivationRange

Function CalculateActivationRange

tensorflow/lite/kernels/kernel_util.h:164–179  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

162// tensor.a
163template <typename T>
164void CalculateActivationRange(TfLiteFusedActivation activation,
165 T* activation_min, T* activation_max) {
166 if (activation == kTfLiteActRelu) {
167 *activation_min = 0;
168 *activation_max = std::numeric_limits<T>::max();
169 } else if (activation == kTfLiteActRelu6) {
170 *activation_min = 0;
171 *activation_max = 6;
172 } else if (activation == kTfLiteActRelu1) {
173 *activation_min = -1;
174 *activation_max = 1;
175 } else {
176 *activation_min = std::numeric_limits<T>::lowest();
177 *activation_max = std::numeric_limits<T>::max();
178 }
179}
180
181// Return true if the given tensors have the same shape.
182bool HaveSameShapes(const TfLiteTensor* input1, const TfLiteTensor* input2);

Callers 14

EvalFloatFunction · 0.85
EvalHybridFunction · 0.85
EvalFloatFunction · 0.85
EvalFloatFunction · 0.85
AverageEvalFloatFunction · 0.85
MaxEvalFloatFunction · 0.85
L2EvalFloatFunction · 0.85
EvalFloatFunction · 0.85
EvalFloatFunction · 0.85
EvalFloatFunction · 0.85
AverageEvalFloatFunction · 0.85
MaxEvalFloatFunction · 0.85

Calls 1

maxFunction · 0.50

Tested by

no test coverage detected