Searched defs:kernel_bias (Results 1 – 1 of 1) sorted by relevance
253 const std::optional<at::Tensor>& kernel_bias, // additive attention bias in _efficient_attention_backward()810 std::optional<Tensor> kernel_bias; in _scaled_dot_product_efficient_attention_backward_cuda() local