Searched refs:NestedTensor_to_mask (Results 1 – 2 of 2) sorted by relevance
188 attn_mask = NestedTensor_to_mask(query, 2, self.size(2)); in NestedTensor_softmax_dropout_cuda()213 Tensor NestedTensor_to_mask(const Tensor& nt, std::optional<int64_t> mask_dim, std::optional<int64_… in NestedTensor_to_mask() function
53 Tensor NestedTensor_to_mask(const Tensor& nt, std::optional<int64_t> mask_dim, std::optional<int64_…