Searched defs:padded_bias (Results 1 – 2 of 2) sorted by relevance
553 auto padded_bias = at::pad_symint(attn_bias, {c10::SymInt(0), pad_count}); in pad_bias() local585 auto padded_bias = at::pad_symint(attn_bias, {c10::SymInt(0), pad_count}); in pad_last_dim() local
333 const auto padded_bias = padRight(bias, std::nullopt, result0.dim() - 1); in native_group_norm_plumbing() local