Lines Matching refs:output_channels
28 size_t output_channels, in create_fully_connected_nc() argument
65 if (output_channels == 0) { in create_fully_connected_nc()
68 xnn_operator_type_to_string(operator_type), output_channels); in create_fully_connected_nc()
80 if (output_stride < output_channels) { in create_fully_connected_nc()
84 xnn_operator_type_to_string(operator_type), output_stride, output_channels); in create_fully_connected_nc()
102 const size_t n_stride = round_up(output_channels, nr); in create_fully_connected_nc()
117 output_channels, input_channels, in create_fully_connected_nc()
124 1, output_channels, input_channels, in create_fully_connected_nc()
132 fully_connected_op->group_output_channels = output_channels; in create_fully_connected_nc()
194 const size_t output_channels = fully_connected_op->group_output_channels; in setup_fully_connected_nc() local
219 size_t nc = output_channels; in setup_fully_connected_nc()
223 …const size_t max_nc = divide_round_up(output_channels * num_other_tiles, num_threads * target_tile… in setup_fully_connected_nc()
231 fully_connected_op->compute.range[1] = output_channels; in setup_fully_connected_nc()
241 size_t output_channels, in xnn_create_fully_connected_nc_qu8() argument
302 input_channels, output_channels, in xnn_create_fully_connected_nc_qu8()
318 size_t output_channels, in xnn_create_fully_connected_nc_f32() argument
357 input_channels, output_channels, in xnn_create_fully_connected_nc_f32()