Searched defs:sumdim (Results 1 – 3 of 3) sorted by relevance
454 at::IntArrayRef sumdim, in _trilinear()
645 auto sumdim = at::dim_list_to_bitset(sumdim_, total_dim); in _trilinear() local
5028 IntArrayRef sumdim, in _trilinear_backward()