viable/strict/1788906341: [BE] Unify Pooling shape checks across backends / families (#196230)
- PyTorch 近 90 天出现 683 次
- PyTorch 近 90 天第 674 次发版
- 上一次:同一天稍早 · viable/strict/1788904239
发生了什么
Started just max_unpooling check, but then realized there are tons of duplicated logic across numerous pooling checks headers: Move all pooling ops helper functions into PoolingChecks.h Delete duplicate max_unpooling3d_shape_check from MaxUnpooling.cu Refactor inlined shapecheck into max_unpool2d_shape_check used by CPU, CUDA and MPS Make gradOutput a trailing std::optional , so forward callers don't have to pass a…
摘要按规则整理自下方来源原文