mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2026-05-08 09:08:10 +02:00
teflon: Add a name field to pipe_tensor
Signed-off-by: Ioana Ciocoi-Radulescu <ruxandra.radulescu@nxp.com>
This commit is contained in:
parent
3c0a55d93f
commit
33b75f298c
2 changed files with 6 additions and 0 deletions
|
|
@ -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];
|
||||
|
|
|
|||
|
|
@ -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.
|
||||
*/
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue