Searched refs:create_from_tensor (Results 1 – 11 of 11) sorted by relevance
| /external/pytorch/test/distributed/checkpoint/ |
| D | test_planner.py | 79 tensor_properties=TensorProperties_Shard.create_from_tensor(torch.zeros(1)),
102 TensorProperties.create_from_tensor(torch.zeros(1)),
113 TensorProperties.create_from_tensor(torch.zeros(1)),
127 TensorProperties.create_from_tensor(tensor),
311 properties=TensorProperties.create_from_tensor(torch.empty([16])),
|
| D | test_compatibility.py | 30 properties=TensorProperties.create_from_tensor(tensor),
|
| D | test_utils.py | 53 tensor_properties=TensorProperties.create_from_tensor(torch.zeros(1)),
|
| /external/pytorch/torch/jit/_passes/ |
| D | _property_propagation.py | 47 graph_i.setType(TensorType.create_from_tensor(example_i)) # type: ignore[arg-type]
|
| /external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| D | metadata.py | 73 def create_from_tensor(tensor: torch.Tensor) -> "TensorProperties": member in TensorProperties
|
| /external/pytorch/torch/distributed/checkpoint/ |
| D | planner_helpers.py | 88 properties=TensorProperties.create_from_tensor(tensor.to_local()),
112 properties=TensorProperties.create_from_tensor(tensor),
|
| D | metadata.py | 102 def create_from_tensor(tensor: torch.Tensor) -> "TensorProperties": member in TensorProperties
|
| /external/pytorch/torch/distributed/tensor/ |
| D | _shards_wrapper.py | 56 wrapper_properties = TensorProperties.create_from_tensor(local_shards[0])
|
| /external/pytorch/torch/distributed/tensor/examples/ |
| D | torchrec_sharding_example.py | 65 wrapper_properties = TensorProperties.create_from_tensor(local_shards[0])
|
| /external/pytorch/test/jit/ |
| D | test_peephole.py | 598 inp.setType(torch._C.TensorType.create_from_tensor(torch.rand([2, 2])))
|
| /external/pytorch/torch/_C/ |
| D | __init__.pyi.in | 2271 def create_from_tensor(t: Tensor) -> TensorType: ...
|