Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorMath.h9 TORCH_API Tensor NestedTensor_to_padded_tensor_generic(
H A DNestedTensorMath.cpp245 Tensor NestedTensor_to_padded_tensor_generic( in NestedTensor_to_padded_tensor_generic() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cpp216 return NestedTensor_to_padded_tensor_generic(t, padding, output_size); in NestedTensor_to_padded_tensor_cuda()