Home
last modified time | relevance | path

Searched defs:tgt_key_padding_mask (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dtransformer.cpp169 const Tensor& tgt_key_padding_mask, in forward()
334 const Tensor& tgt_key_padding_mask, in forward()
427 const Tensor& tgt_key_padding_mask, in forward()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp1459 torch::Tensor tgt_key_padding_mask = in transformer_test_helper() local