You can allocate memory on PSRAM for your Tensor Arena
if (tensor_arena == NULL) {
//allocate memory for TensorArena on PSRAM
tensor_arena = (uint8_t *) ps_malloc(kTensorArenaSize);
}
This code shows how to use ESP32's built-in PSRAM in tensorflow-lite-micro Person Detection example.
C++MIT
You can allocate memory on PSRAM for your Tensor Arena
if (tensor_arena == NULL) {
//allocate memory for TensorArena on PSRAM
tensor_arena = (uint8_t *) ps_malloc(kTensorArenaSize);
}