AlexIndustrial/mesa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

teflon: Set unused dimensions to 1

Browse Source 
So drivers don't need to superfluously watch out for zeroes when
multiplying among dimensions.

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/34498>
This commit is contained in:
Tomeu Vizoso
2025-04-10 17:03:17 +02:00
committed by Marge Bot
parent 23ae1c3bff
commit c4a5f8d665
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/gallium/frontends/teflon/tfl_device.c
+7 -1
View File
@@ -223,7 +223,13 @@ fill_tensor(struct teflon_delegate *delegate, TfLiteContext *tf_context, struct
tensor->resource = create_resource(context, tf_tensor);
tensor->index = index;
memcpy(tensor->dims, tf_tensor.dims->data, tf_tensor.dims->size * sizeof(*tensor->dims));
for (int out_dim = 0; out_dim < 4; out_dim++) {
int in_dim = tf_tensor.dims->size - 4 + out_dim;
if (in_dim >= 0)
tensor->dims[out_dim] = tf_tensor.dims->data[in_dim];
else
tensor->dims[out_dim] = 1;
}
if (tf_tensor.quantization.type == kTfLiteAffineQuantization) {
const TfLiteAffineQuantization *quant = (const TfLiteAffineQuantization *)tf_tensor.quantization.params;