Searched refs:cascade_sum_num_levels (Results 1 – 2 of 2) sorted by relevance
185 constexpr int64_t cascade_sum_num_levels = 8; in nll_loss2d_forward_out_frame() local187 scalar_t weight_partial_sums[cascade_sum_num_levels] = {0}; in nll_loss2d_forward_out_frame()189 scalar_t loss_partial_sums[cascade_sum_num_levels] = {0}; in nll_loss2d_forward_out_frame()191 std::max(int64_t(4), utils::CeilLog2(numiter) / cascade_sum_num_levels); in nll_loss2d_forward_out_frame()220 for (int64_t j = 0; j + 1 < cascade_sum_num_levels; ++j) { in nll_loss2d_forward_out_frame()
232 constexpr int64_t cascade_sum_num_levels = 8; in nll_loss_out_frame() local234 std::max(int64_t(4), utils::CeilLog2(batch_size) / cascade_sum_num_levels); in nll_loss_out_frame()241 scalar_t weight_partial_sums[cascade_sum_num_levels] = {0}; in nll_loss_out_frame()243 scalar_t loss_partial_sums[cascade_sum_num_levels] = {0}; in nll_loss_out_frame()266 for (int64_t j = 0; j + 1 < cascade_sum_num_levels; ++j) { in nll_loss_out_frame()