Searched refs:NestedTensor_get_max_size (Results 1 – 6 of 6) sorted by relevance
55 std::vector<int64_t> NestedTensor_get_max_size(const NestedTensorImpl& nt) { in NestedTensor_get_max_size() function
233 …const auto result_size_1 = mask_dim_length ? *mask_dim_length : NestedTensor_get_max_size(*nt_impl… in NestedTensor_to_mask()
130 TORCH_API std::vector<int64_t> NestedTensor_get_max_size(
256 auto max_size = NestedTensor_get_max_size(nt); in NestedTensor_to_padded_tensor_generic()
169 auto new_size = NestedTensor_get_max_size(*nt_input); in NestedTensor_to_padded_tensor_cuda()
386 ? NestedTensor_get_max_size(*get_nested_tensor_impl(qkv))[0] in transform_bias_rescale_qkv_cuda()