Remove TensorStorage Class
TensorStorage class is used for ForwardCheck Operations.
Refactor can allow to use ctx.save_for_backward in a more pythonic way.
-
Call forward operation "recursively" in Checkpointable class -
Call backward operation "recursively" in CheckpointOptim class.