Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cpp139 Tensor NestedTensor_to_padded_tensor_cuda( in NestedTensor_to_padded_tensor_cuda() function
163 NestedTensor_to_padded_tensor_cuda(result, padding, output_size); in NestedTensor_to_padded_tensor_cuda()