Searched refs:pad_tensor_to_shape (Results 1 – 1 of 1) sorted by relevance
39 Tensor pad_tensor_to_shape( in pad_tensor_to_shape() function237 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()