Home
last modified time | relevance | path

Searched refs:nested_from_padded_generic (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/nested/cuda/
DNestedTensorTransformerFunctions.cpp50 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()
/external/pytorch/aten/src/ATen/native/nested/
DNestedTensorMatmul.cpp304 auto output_nested = nested_from_padded_generic(output_padded, output_sizes); in matmul_nested()
DNestedTensorMath.cpp204 Tensor nested_from_padded_generic( in nested_from_padded_generic() function