Searched defs:num_other_tiles (Results 1 – 3 of 3) sorted by relevance
221 const size_t num_other_tiles = divide_round_up(batch_size, mr); in setup_fully_connected_nc() local
551 const size_t num_other_tiles = groups * batch_size * divide_round_up(output_size, mr); in setup_conv_path() local728 const size_t num_other_tiles = groups * stride_height * stride_width * in setup_subconv2d_path() local
883 const size_t num_other_tiles = groups * divide_round_up(batch_output_size, mr); in setup_convolution2d_nhwc() local996 const size_t num_other_tiles = groups * batch_size * divide_round_up(output_size, mr); in setup_convolution2d_nhwc() local