def METHOD_NAME(policy_name: str) -> Type[TensorPlacementPolicy]:
    if policy_name == 'cpu':
        return CPUTensorPlacementPolicy
    elif policy_name == 'cuda':
        return CUDATensorPlacementPolicy
    elif policy_name == 'auto':
        return AutoTensorPlacementPolicy
    else:
        raise TypeError(f"Unknown tensor placement policy {policy_name}")