Searched refs:_create_write_item_for_tensor (Results 1 – 2 of 2) sorted by relevance
8 from torch.distributed.checkpoint.planner_helpers import _create_write_item_for_tensor15 write_item_1 = _create_write_item_for_tensor("tensor_0", torch.rand(4))19 write_item_2 = _create_write_item_for_tensor(second_fqn, torch.rand(10))
105 def _create_write_item_for_tensor(fqn: str, tensor: torch.Tensor) -> WriteItem: function215 requests.append(_create_write_item_for_tensor(fqn, obj))235 return [_create_write_item_for_tensor(fqn, object)]