Searched refs:idx_large_norm (Results 1 – 1 of 1) sorted by relevance
2682 auto idx_large_norm = (norm_cpu >= thetas[total_n_degs - 2]) in mexp_impl() local2685 if (idx_large_norm.numel()) { in mexp_impl()2687 idx_large_norm, a in mexp_impl()2696 res.index_put_({idx_large_norm}, mexp_out); in mexp_impl()