[feat] support custom segment_size for expandable_segments-pytorch-AtomGit

Star845

Star845

已合并

[feat] support custom segment_size for expandable_segments #26374

XDaoHong创建于 2025年11月10日从XDaoHong/pytorch： v2.9.0 合入到 Ascend/pytorch： v2.9.0

[feat] support custom segment_size for expandable_segments #26374

已合并

XDaoHong创建于 2025年11月10日

从XDaoHong/pytorch v2.9.0合入到Ascend/pytorch v2.9.0