Searched defs:attn_bias_ (Results 1 – 2 of 2) sorted by relevance
565 std::optional<Tensor> attn_bias_; in _scaled_dot_product_efficient_attention_batch_rule() local606 std::optional<Tensor> attn_bias_; in _scaled_dot_product_cudnn_attention_batch_rule() local
203 std::optional<Tensor> attn_bias_; in _scaled_dot_product_cudnn_attention_backward_cuda() local