diff --git a/src/gallium/frontends/teflon/tfl_device.c b/src/gallium/frontends/teflon/tfl_device.c index 1d27639a56b..e31e1215d3a 100644 --- a/src/gallium/frontends/teflon/tfl_device.c +++ b/src/gallium/frontends/teflon/tfl_device.c @@ -231,6 +231,9 @@ fill_operation(struct teflon_delegate *delegate, TfLiteContext *tf_context, TfLi case kTfLiteBuiltinAbs: operation->type = PIPE_ML_OPERATION_TYPE_ABSOLUTE; break; + case kTfLiteBuiltinLogistic: + operation->type = PIPE_ML_OPERATION_TYPE_LOGISTIC; + break; default: return false; } @@ -370,6 +373,9 @@ dump_graph(struct pipe_tensor *tensors, unsigned tensor_count, struct pipe_ml_op case PIPE_ML_OPERATION_TYPE_ABSOLUTE: teflon_debug("%-6s ", "ABS"); break; + case PIPE_ML_OPERATION_TYPE_LOGISTIC: + teflon_debug("%-6s ", "LOG"); + break; } for (unsigned j = 0; j < operations[i].input_count; j++) { diff --git a/src/gallium/include/pipe/p_state.h b/src/gallium/include/pipe/p_state.h index b6df901b133..699a5d94e70 100644 --- a/src/gallium/include/pipe/p_state.h +++ b/src/gallium/include/pipe/p_state.h @@ -1051,6 +1051,7 @@ enum pipe_ml_operation_type { PIPE_ML_OPERATION_TYPE_RESHAPE, PIPE_ML_OPERATION_TYPE_RELU, PIPE_ML_OPERATION_TYPE_ABSOLUTE, + PIPE_ML_OPERATION_TYPE_LOGISTIC, }; /**