fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2025-12-20 07:20:10 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

teflon: Add support for the ResizeNearestNeighbor operation

Browse source 
Acked-by: Christian Gmeiner <cgmeiner@igalia.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36699>
This commit is contained in:
Tomeu Vizoso 2025-08-09 10:38:17 +02:00 • committed by Marge Bot
parent 0001dab219
commit b3262b37ce
2 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/gallium/frontends/teflon/tfl_device.c
View file

@ -269,6 +269,10 @@ fill_operation(struct teflon_delegate *delegate, TfLiteContext *tf_context, TfLi
break; break;
} }
case kTfLiteBuiltinResizeNearestNeighbor: {
operation->type = PIPE_ML_OPERATION_TYPE_RESIZE;
break;
}
default: default:
return false; return false;
} }
@ -420,6 +424,9 @@ dump_graph(struct pipe_tensor *tensors, unsigned tensor_count, struct pipe_ml_op
case PIPE_ML_OPERATION_TYPE_STRIDED_SLICE: case PIPE_ML_OPERATION_TYPE_STRIDED_SLICE:
teflon_debug("%-6s ", "STRIDED_SLICE"); teflon_debug("%-6s ", "STRIDED_SLICE");
break; break;
case PIPE_ML_OPERATION_TYPE_RESIZE:
teflon_debug("%-6s ", "RESIZE");
break;
} }
for (unsigned j = 0; j < operations[i].input_count; j++) { for (unsigned j = 0; j < operations[i].input_count; j++) {
@ -629,6 +636,8 @@ tflite_builtin_op_name(TfLiteBuiltinOperator op)
return "MEAN"; return "MEAN";
case kTfLiteBuiltinStridedSlice: case kTfLiteBuiltinStridedSlice:
return "STRIDED_SLICE"; return "STRIDED_SLICE";
case kTfLiteBuiltinResizeNearestNeighbor:
return "RESIZE";
default: default:
return "unknown"; return "unknown";
} }

1
src/gallium/include/pipe/p_state.h
View file

@ -1055,6 +1055,7 @@ enum pipe_ml_operation_type {
PIPE_ML_OPERATION_TYPE_SUBTRACT, PIPE_ML_OPERATION_TYPE_SUBTRACT,
PIPE_ML_OPERATION_TYPE_TRANSPOSE, PIPE_ML_OPERATION_TYPE_TRANSPOSE,
PIPE_ML_OPERATION_TYPE_STRIDED_SLICE, PIPE_ML_OPERATION_TYPE_STRIDED_SLICE,
PIPE_ML_OPERATION_TYPE_RESIZE,
}; };
enum pipe_ml_pooling_type { enum pipe_ml_pooling_type {