teflon: Add a name field to pipe_tensor

Signed-off-by: Ioana Ciocoi-Radulescu <ruxandra.radulescu@nxp.com>
This commit is contained in:
Ioana Ciocoi-Radulescu 2026-04-24 12:38:01 +03:00
parent 3c0a55d93f
commit 33b75f298c
2 changed files with 6 additions and 0 deletions

View file

@ -408,6 +408,8 @@ fill_tensor(struct teflon_delegate *delegate, TfLiteContext *tf_context, struct
tensor->dims[out_dim] = 1;
}
tensor->name = strdup(tf_tensor.name);
if (tf_tensor.quantization.type == kTfLiteAffineQuantization) {
const TfLiteAffineQuantization *quant = (const TfLiteAffineQuantization *)tf_tensor.quantization.params;
tensor->scale = quant->scale->data[0];

View file

@ -1039,6 +1039,10 @@ struct pipe_tensor {
* Zero-points used to quantize this tensor, per-axis quantization.
*/
int *zero_points;
/**
* Tensor name
*/
const char *name;
/**
* Whether the tensor contains data in INT8 or UINT8 format.
*/