Lines Matching refs:output_channels
28 size_t output_channels, in xnn_create_fully_connected_nc_q8() argument
61 if (output_channels == 0) { in xnn_create_fully_connected_nc_q8()
64 output_channels); in xnn_create_fully_connected_nc_q8()
76 if (output_stride < output_channels) { in xnn_create_fully_connected_nc_q8()
80 output_stride, output_channels); in xnn_create_fully_connected_nc_q8()
135 const uint32_t n_stride = round_up(output_channels, nr); in xnn_create_fully_connected_nc_q8()
148 output_channels, input_channels, in xnn_create_fully_connected_nc_q8()
155 1, output_channels, input_channels, in xnn_create_fully_connected_nc_q8()
163 fully_connected_op->group_output_channels = output_channels; in xnn_create_fully_connected_nc_q8()
196 size_t output_channels, in xnn_create_fully_connected_nc_f32() argument
223 if (output_channels == 0) { in xnn_create_fully_connected_nc_f32()
226 output_channels); in xnn_create_fully_connected_nc_f32()
238 if (output_stride < output_channels) { in xnn_create_fully_connected_nc_f32()
242 output_stride, output_channels); in xnn_create_fully_connected_nc_f32()
277 const uint32_t n_stride = round_up(output_channels, nr); in xnn_create_fully_connected_nc_f32()
290 output_channels, input_channels, in xnn_create_fully_connected_nc_f32()
296 1, output_channels, input_channels, in xnn_create_fully_connected_nc_f32()
303 fully_connected_op->group_output_channels = output_channels; in xnn_create_fully_connected_nc_f32()
364 const size_t output_channels = fully_connected_op->group_output_channels; in setup_fully_connected_nc() local
389 size_t nc = output_channels; in setup_fully_connected_nc()
393 …const size_t max_nc = divide_round_up(output_channels * num_other_tiles, num_threads * target_tile… in setup_fully_connected_nc()
401 fully_connected_op->compute.range[1] = output_channels; in setup_fully_connected_nc()