AvgPool1D算子支持Ascend950
AvgPool1D是一维平均池化算子,采用AscendC SIMT实现,支持Grid-Stride多核并行+单线程内顺序确定性累加,不使用原子操作。支持不对称padding、ceil_mode输出长度修正和count_include_pad分母选择。
描述
AvgPool1D算子支持Ascend950
AvgPool1D是一维平均池化算子,采用AscendC SIMT实现,支持Grid-Stride多核并行+单线程内顺序确定性累加,不使用原子操作。支持不对称padding、ceil_mode输出长度修正和count_include_pad分母选择。
测试
文档更新
类型标签