Searched refs:num_other_tiles (Results 1 – 3 of 3) sorted by relevance
391 const size_t num_other_tiles = divide_round_up(batch_size, mr); in setup_fully_connected_nc() local393 …const size_t max_nc = divide_round_up(output_channels * num_other_tiles, num_threads * target_tile… in setup_fully_connected_nc()
622 const size_t num_other_tiles = groups * batch_size * divide_round_up(output_size, mr); in setup_conv_path() local624 …const size_t max_nc = divide_round_up(group_output_channels * num_other_tiles, num_threads * targe… in setup_conv_path()743 const size_t num_other_tiles = groups * stride_height * stride_width * in setup_subconv2d_path() local746 …const size_t max_nc = divide_round_up(group_output_channels * num_other_tiles, num_threads * targe… in setup_subconv2d_path()
828 const size_t num_other_tiles = groups * divide_round_up(batch_output_size, mr); in setup_convolution2d_nhwc() local830 …const size_t max_nc = divide_round_up(group_output_channels * num_other_tiles, num_threads * targe… in setup_convolution2d_nhwc()919 const size_t num_other_tiles = groups * batch_size * divide_round_up(output_size, mr); in setup_convolution2d_nhwc() local921 …const size_t max_nc = divide_round_up(group_output_channels * num_other_tiles, num_threads * targe… in setup_convolution2d_nhwc()