Searched refs:decoder_state_tensor (Results 1 – 2 of 2) sorted by relevance
187 decoder_state_tensor = torch.from_numpy(decoder_state).to(210 _Q = decoder_state_tensor.unsqueeze(1).transpose(0, 1)
3563 const torch::Tensor decoder_state_tensor = decoder_state; in _multihead_attn_test_helper() local3603 torch::Tensor _Q = decoder_state_tensor.unsqueeze(1).transpose(0, 1); in _multihead_attn_test_helper()