result = rewriter.create<mhlo::DivOp>(loc, pooled_type, pooled,
                                          divisor.getResult(0));
  }
  return result;
}

Value GetAvgPoolInput(TF::AvgPoolOp op) { return op.getValue(); }
Value GetAvgPoolInput(TF::AvgPool3DOp op) { return op.getInput(); }

// Converts AvgPool op to HLO ReduceWindow op by setting appropriate window