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

Function MaxEvalFloat

tensorflow/lite/experimental/micro/kernels/pooling.cc:96–115  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

94}
95
96void MaxEvalFloat(TfLiteContext* context, TfLiteNode* node,
97 TfLitePoolParams* params, OpData* data,
98 const TfLiteTensor* input, TfLiteTensor* output) {
99 float activation_min, activation_max;
100 CalculateActivationRange(params->activation, &activation_min,
101 &activation_max);
102
103 tflite::PoolParams op_params;
104 op_params.stride_height = params->stride_height;
105 op_params.stride_width = params->stride_width;
106 op_params.filter_height = params->filter_height;
107 op_params.filter_width = params->filter_width;
108 op_params.padding_values.height = data->padding.height;
109 op_params.padding_values.width = data->padding.width;
110 op_params.float_activation_min = activation_min;
111 op_params.float_activation_max = activation_max;
112 reference_ops::MaxPool(op_params, GetTensorShape(input),
113 GetTensorData<float>(input), GetTensorShape(output),
114 GetTensorData<float>(output));
115}
116
117void MaxEvalQuantizedUInt8(TfLiteContext* context, TfLiteNode* node,
118 TfLitePoolParams* params, OpData* data,

Callers 1

MaxEvalFunction · 0.70

Calls 3

CalculateActivationRangeFunction · 0.85
MaxPoolClass · 0.50
GetTensorShapeFunction · 0.50

Tested by

no test coverage detected