Create a separate callback for restored tensor propagation and move it out of LoadRestoredTensorAsIfrtLoadedVariable.
Create a separate callback for restored tensor propagation and move it out of LoadRestoredTensorAsIfrtLoadedVariable.
