Searched refs:batch_output_size (Results 1 – 1 of 1) sorted by relevance
849 const size_t batch_output_size = batch_size * output_size; in setup_convolution2d_nhwc() local860 …if (batch_output_size == 1 && convolution_op->ukernel.gemm.mr1_case.function[XNN_UARCH_DEFAULT] !=… in setup_convolution2d_nhwc()883 const size_t num_other_tiles = groups * divide_round_up(batch_output_size, mr); in setup_convolution2d_nhwc()903 convolution_op->compute.range[0] = batch_output_size; in setup_convolution2d_nhwc()921 convolution_op->compute.range[1] = batch_output_size; in setup_convolution2d_nhwc()1143 …const size_t batch_output_size = batch_size * convolution_op->output_height * convolution_op->outp… in setup_convolution2d_nhwc() local1156 size_t mc = batch_output_size; in setup_convolution2d_nhwc()1159 … const size_t max_mc = divide_round_up(batch_output_size, num_threads * target_tiles_per_thread); in setup_convolution2d_nhwc()1167 convolution_op->compute.range[0] = batch_output_size; in setup_convolution2d_nhwc()