Searched refs:batch_range (Results 1 – 2 of 2) sorted by relevance
| /external/XNNPACK/src/ |
| D | operator-run.c | 513 size_t batch_range) in xnn_compute_prelu() argument
520 context->ukernel(batch_range, context->n, x, x_stride, context->w, y, y_stride, &context->params); in xnn_compute_prelu()
526 size_t batch_range) in xnn_compute_channel_pad() argument
533 …context->ukernel(batch_range, context->n, context->l, context->r, context->c, x, x_stride, y, y_st… in xnn_compute_channel_pad()
539 size_t batch_range /* always 1 */) in xnn_compute_add_strided() argument
541 assert(batch_range == 1); in xnn_compute_add_strided()
626 size_t batch_range /* always 1 */) in xnn_compute_univector_strided() argument
628 assert(batch_range == 1); in xnn_compute_univector_strided()
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 553 size_t batch_range);
668 size_t batch_range);
706 size_t batch_range);
745 size_t batch_range);
|