ESP32 ML - C Optimization for Input Normalization

mlops.c

STATIC mp_obj_t norm_input(mp_obj_t input, mp_obj_t multiplier, mp_obj_t adder) {
  // Check if input is a list
  if (mp_obj_is_type(input, &mp_type_list)) {

    // Create a new list to hold normalized values
    mp_obj_t outputs = mp_obj_new_list(0, NULL);
	
	// Convert input list to pointer and get length
	mp_obj_list_t *input_c = MP_OBJ_TO_PTR(input);
	unsigned int list_len = input_c->len;

    for (int i = 0; i < list_len; i++) {
		// relu each item in input list
      mp_obj_t normalized = norm_input(input_c->items[i], multiplier, adder);
		// append to new list
	  mp_obj_list_append(outputs, normalized);
    }

    return outputs;

  } else {

    // relu the scalar input 
    float input_f = mp_obj_get_float(input);
	float multiplier_f =  mp_obj_get_float(multiplier);
	float adder_f = mp_obj_get_float(adder);
    return mp_obj_new_float((input_f+adder_f)*multiplier_f);

  }

}