Searched refs:max_reduce_threads (Results 1 – 1 of 1) sorted by relevance
70 constexpr int max_reduce_threads(c10::ScalarType type) { in max_reduce_threads() function924 max_reduce_threads(desc.f_inputs_type) / config.output_vec_size; in launch_jitted_reduce_kernel()