: public impl::ReplicateTensorListInitOpsPassBase<
          ReplicateTensorListInitOps> {
 public:
  void runOnOperation() override {
    getOperation().walk([](Operation* op) {
      if (auto tl_reserve = dyn_cast<TensorListReserveOp>(op)) {
        ReplicateTensorListForUses(tl_reserve);
      }
      if (auto tl_empty = dyn_cast<EmptyTensorListOp>(op)) {
        ReplicateTensorListForUses(tl_empty);
      }
    });
  }
};