Searched refs:nested_from_padded_generic (Results 1 – 3 of 3) sorted by relevance
50 return at::native::nested_from_padded_generic(padded, sizes, do_transform_0213); in nested_from_padded_cuda()56 return at::native::nested_from_padded_generic(padded, sizes, do_transform_0213); in nested_from_padded_cuda()118 return at::native::nested_from_padded_generic(padded, sizes); in nested_from_padded_cuda()
304 auto output_nested = nested_from_padded_generic(output_padded, output_sizes); in matmul_nested()
204 Tensor nested_from_padded_generic( in nested_from_padded_generic() function