Lines Matching defs:batch_index
346 size_t batch_index, in xnn_compute_spmm()
364 size_t batch_index, in xnn_compute_grouped_batch_igemm()
417 size_t batch_index, in xnn_compute_batch_igemm()
468 size_t batch_index, in xnn_compute_grouped_subgemm2d()
506 size_t batch_index, in xnn_compute_subgemm2d()
543 size_t batch_index, in xnn_compute_grouped_subconv2d()
582 size_t batch_index, in xnn_compute_subconv2d()
620 size_t batch_index, in xnn_compute_conv2d_hwc2chw()
642 size_t batch_index, in xnn_compute_dwconv_unipass()
661 size_t batch_index, in xnn_compute_dwconv2d_chw()
677 size_t batch_index, in xnn_compute_argmax_pooling_unipass()
696 size_t batch_index, in xnn_compute_argmax_pooling_multipass()
718 size_t batch_index, in xnn_compute_max_pooling()
756 size_t batch_index, in xnn_compute_average_pooling_unipass()
774 size_t batch_index, in xnn_compute_average_pooling_multipass()
795 size_t batch_index, in xnn_compute_pixelwise_average_pooling_unipass()
815 size_t batch_index, in xnn_compute_pixelwise_average_pooling_multipass()
837 size_t batch_index) in xnn_compute_global_average_pooling_nwc_unipass()
856 size_t batch_index) in xnn_compute_global_average_pooling_nwc_multipass()
879 size_t batch_index, in xnn_compute_global_average_pooling_ncw()
898 size_t batch_index, in xnn_compute_resize_bilinear()
917 size_t batch_index, in xnn_compute_resize_bilinear_chw()
1063 size_t batch_index) in xnn_compute_lut_strided()
1084 size_t batch_index, in xnn_compute_univector_strided()
1113 size_t batch_index) in xnn_compute_u8_softmax()
1128 size_t batch_index) in xnn_compute_floating_point_softmax()
1231 size_t batch_index, in xnn_compute_hmp_grouped_batch_igemm()
1286 size_t batch_index, in xnn_compute_batch_hmp_igemm()