MCPcopy Create free account
hub / github.com/Tiiny-AI/PowerInfer / ggml_compute_forward_unary

Function ggml_compute_forward_unary

ggml.c:13241–13297  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

13239//gmml_compute_forward_unary
13240
13241static void ggml_compute_forward_unary(
13242 const struct ggml_compute_params * params,
13243 const struct ggml_tensor * src0,
13244 struct ggml_tensor * dst) {
13245 const enum ggml_unary_op op = ggml_get_unary_op(dst);
13246
13247 switch (op) {
13248 case GGML_UNARY_OP_ABS:
13249 {
13250 ggml_compute_forward_abs(params, src0, dst);
13251 } break;
13252 case GGML_UNARY_OP_SGN:
13253 {
13254 ggml_compute_forward_sgn(params, src0, dst);
13255 } break;
13256 case GGML_UNARY_OP_NEG:
13257 {
13258 ggml_compute_forward_neg(params, src0, dst);
13259 } break;
13260 case GGML_UNARY_OP_STEP:
13261 {
13262 ggml_compute_forward_step(params, src0, dst);
13263 } break;
13264 case GGML_UNARY_OP_TANH:
13265 {
13266 ggml_compute_forward_tanh(params, src0, dst);
13267 } break;
13268 case GGML_UNARY_OP_ELU:
13269 {
13270 ggml_compute_forward_elu(params, src0, dst);
13271 } break;
13272 case GGML_UNARY_OP_RELU:
13273 {
13274 ggml_compute_forward_relu(params, src0, dst);
13275 } break;
13276 case GGML_UNARY_OP_GELU:
13277 {
13278 ggml_compute_forward_gelu(params, src0, dst);
13279 } break;
13280 case GGML_UNARY_OP_GELU_QUICK:
13281 {
13282 ggml_compute_forward_gelu_quick(params, src0, dst);
13283 } break;
13284 case GGML_UNARY_OP_SILU:
13285 {
13286 ggml_compute_forward_silu(params, src0, dst);
13287 } break;
13288 case GGML_UNARY_OP_LEAKY:
13289 {
13290 ggml_compute_forward_leaky(params, src0, dst);
13291 } break;
13292 default:
13293 {
13294 GGML_ASSERT(false);
13295 } break;
13296 }
13297}
13298

Callers 1

ggml_compute_forwardFunction · 0.70

Calls 12

ggml_get_unary_opFunction · 0.70
ggml_compute_forward_absFunction · 0.70
ggml_compute_forward_sgnFunction · 0.70
ggml_compute_forward_negFunction · 0.70
ggml_compute_forward_eluFunction · 0.70

Tested by

no test coverage detected