Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorMath.cpp39 Tensor pad_tensor_to_shape( in pad_tensor_to_shape() function
237 masks.push_back(pad_tensor_to_shape(mask_i, target_size_arr)); in nested_from_padded_generic()
307 buffers.push_back(pad_tensor_to_shape(to_pad, max_size, padding)); in NestedTensor_to_padded_tensor_generic()
323 return pad_tensor_to_shape(ret_val, output_size_, padding); in NestedTensor_to_padded_tensor_generic()