ITADN

Batch an MLIR Region

#2792Openjoelberkeley 创建于 2026-04-22
J
joelberkeleycommented
I can AD an MLIR Region with `AutoDiffRegionOp`. Can I similarly batch an MLIR Region? I only see `BatchOp`, which operates on function names afaict (the `summary` is a copy-paste from `AutoDiffOp`). I specifically want this for StableHLO.
2 条评论