PatternRewriter &rewriter) const override {
    Location loc = op.getLoc();
    Value grad = op.getYBackprop();
    Value act = op.getX();
    Value scale = op.getScale();
    Value mean = op.getReserveSpace_1();
    Value var = op.getReserveSpace_2();

    // TODO(b/141785544): Update this to not require static shapes.