Home
last modified time | relevance | path

Searched refs:floating_point_softmax_context (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/XNNPACK/src/xnnpack/
H A Dcompute.h1084 struct floating_point_softmax_context { struct
1106 const struct floating_point_softmax_context context[restrict XNN_MIN_ELEMENTS(1)], argument
H A Doperator.h294 struct floating_point_softmax_context floating_point_softmax;
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dsoftmax-nc.c335 softmax_op->context.floating_point_softmax = (struct floating_point_softmax_context) { in setup_softmax_nc_floating_point()
/aosp_15_r20/external/XNNPACK/src/
H A Doperator-run.c1127 const struct floating_point_softmax_context context[restrict XNN_MIN_ELEMENTS(1)], in xnn_compute_floating_point_softmax()