Searched defs:expand2 (Results 1 – 3 of 3) sorted by relevance
452 at::IntArrayRef expand2, in _trilinear()
643 auto expand2 = at::dim_list_to_bitset(expand2_, total_dim); in _trilinear() local
5026 IntArrayRef expand2, in _trilinear_backward()