Searched defs:expand3 (Results 1 – 3 of 3) sorted by relevance
453 at::IntArrayRef expand3, in _trilinear()
644 auto expand3 = at::dim_list_to_bitset(expand3_, total_dim); in _trilinear() local
5027 IntArrayRef expand3, in _trilinear_backward()