parameters: (tensor,
                    device,
                    assign_tuple_sharding=False,
                    use_sharding_op=False)
return_statement: return Sharding.assign_device(device).apply_to_tensor(
        tensor,
        assign_tuple_sharding=assign_tuple_sharding,
        use_sharding_op=use_sharding_op)
docstring: Returns a tensor that has AssignDevice sharding attribute.
docstring_summary: Returns a tensor that has AssignDevice sharding attribute.
docstring_tokens: Returns a tensor that has AssignDevice sharding attribute .
function: def assign_device(tensor,
                    device,
                    assign_tuple_sharding=False,
                    use_sharding_op=False):
    """Returns a tensor that has AssignDevice sharding attribute."""
    return Sharding.assign_device(device).apply_to_tensor(
        tensor,
        assign_tuple_sharding=assign_tuple_sharding,
        use_sharding_op=use_sharding_op)
function_tokens: def assign_device ( tensor , device , assign_tuple_sharding = False , use_sharding_op = False ) : return Sharding . assign_device ( device ) . apply_to_tensor ( tensor , assign_tuple_sharding = assign_tuple_sharding , use_sharding_op = use_sharding_op )