src_key_padding_mask (reference) in projects: aosp_15_r20

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

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

/aosp_15_r20/external/pytorch/torch/nn/modules/
H A D	transformer.py	179 src_key_padding_mask: Optional[Tensor] = None, 275 src_key_padding_mask=src_key_padding_mask, 388 src_key_padding_mask: Optional[Tensor] = None, 408 src_key_padding_mask = F._canonical_mask( 409 mask=src_key_padding_mask, 428 src_key_padding_mask_for_layers = src_key_padding_mask 450 elif src_key_padding_mask is None: 455 src, src_key_padding_mask.logical_not() 500 if (not why_not_sparsity_fast_path) and (src_key_padding_mask is not None): 503 output, src_key_padding_mask.logical_not(), mask_check=False [all …]
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A D	transformer.cpp	`66 const Tensor& src_key_padding_mask) { in forward() argument 69 src, src, src, src_key_padding_mask, /need_weights=/true, src_mask)); in forward() 265 const Tensor& src_key_padding_mask) { in forward() argument 270 src, src_mask, src_key_padding_mask); in forward() 274 output, src_mask, src_key_padding_mask); in forward() 426 const Tensor& src_key_padding_mask, in forward() argument 453 this->encoder.forward<Tensor>(src, src_mask, src_key_padding_mask); in forward()`
/aosp_15_r20/external/pytorch/test/
H A D	test_transformers.py	235 outputs = encoder(inputs, src_key_padding_mask=pad_mask) 247 test_train_uint8 = encoder(test, src_key_padding_mask=pad_mask.to(torch.uint8)) 252 test_train_bool = encoder(test, src_key_padding_mask=pad_mask) 258 test_eval_uint8 = encoder(test, src_key_padding_mask=pad_mask.to(torch.int64)) 263 test_eval_bool = encoder(test, src_key_padding_mask=pad_mask) 393 for input, src_key_padding_mask in input_mask_pairs: 395 fastpath_output = model(input, src_key_padding_mask=src_key_padding_mask) 396 … slowpath_output = model(input, src_key_padding_mask=src_key_padding_mask) # reference 402 expanded_seqlen = src_key_padding_mask.shape[1] 406 …fastpath_output_expanded = fastpath_output_expanded.masked_fill(src_key_padding_mask.unsqueeze(-1)… [all …]
H A D	test_nn.py	2841 src_key_padding_mask = torch.rand(bsz, seq_length) >= 0.5 2849 src_key_padding_mask=src_key_padding_mask, 3567 src_key_padding_mask = torch.rand(src_key_padding_mask_size) >= 0.5 3569 src_key_padding_mask = None 3586 src_key_padding_mask=src_key_padding_mask, 9917 …src_key_padding_mask = src_key_padding_mask_orig.reshape(B, 1, 1, L).expand(B, num_heads, L, L).bo… 9922 (src_key_padding_mask_orig, src_key_padding_mask, 1), 9968 src_key_padding_mask = torch.randint(0, 2, (B, L)).bool() 9971 masks = [(src_mask, 0), (src_key_padding_mask, 1), (generic_mask, 2)] 12437 result = model(encoder_input, src_key_padding_mask=mask) [all …]
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A D	transformer.h	`93 const Tensor& src_key_padding_mask = {},`
H A D	transformercoder.h	`49 const Tensor& src_key_padding_mask = {});`
H A D	transformerlayer.h	`47 const Tensor& src_key_padding_mask = {});`
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A D	common_modules.py	2453 for src_mask, src_key_padding_mask, norm_first, batch_first, bias in \ 2461 make_input((3, 4)), src_mask=src_mask, src_key_padding_mask=src_key_padding_mask 2575 src_key_padding_mask, tgt_key_padding_mask = (key_padding_mask,) * 2 2583 … tgt_key_padding_mask=tgt_key_padding_mask, src_key_padding_mask=src_key_padding_mask 2596 src_key_padding_mask, tgt_key_padding_mask = (key_padding_mask.expand(2, 3),) * 2 2605 … tgt_key_padding_mask=tgt_key_padding_mask, src_key_padding_mask=src_key_padding_mask
/aosp_15_r20/external/pytorch/test/dynamo/
H A D	test_repros.py	`881 def forward(self, src, src_mask=None, src_key_padding_mask=None): argument 883 x = self.norm1(x + self._sa_block(x, src_mask, src_key_padding_mask))`