clk: clk: Add clk_gate_restore_context function