from .stateful_tensor_mgr import StatefulTensorMgr from .tensor_placement_policy import TensorPlacementPolicyFactory __all__ = ['StatefulTensorMgr', 'TensorPlacementPolicyFactory']