Home
last modified time | relevance | path

Searched refs:create_from_tensor (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/checkpoint/
H A Dtest_planner.py79 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])),
H A Dtest_compatibility.py30 properties=TensorProperties.create_from_tensor(tensor),
H A Dtest_utils.py53 tensor_properties=TensorProperties.create_from_tensor(torch.zeros(1)),
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dplanner_helpers.py88 properties=TensorProperties.create_from_tensor(tensor.to_local()),
112 properties=TensorProperties.create_from_tensor(tensor),
H A Dmetadata.py102 def create_from_tensor(tensor: torch.Tensor) -> "TensorProperties": member in TensorProperties
/aosp_15_r20/external/pytorch/torch/jit/_passes/
H A D_property_propagation.py47 graph_i.setType(TensorType.create_from_tensor(example_i)) # type: ignore[arg-type]
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/
H A Dmetadata.py73 def create_from_tensor(tensor: torch.Tensor) -> "TensorProperties": member in TensorProperties
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_shards_wrapper.py56 wrapper_properties = TensorProperties.create_from_tensor(local_shards[0])
/aosp_15_r20/external/pytorch/torch/distributed/tensor/examples/
H A Dtorchrec_sharding_example.py65 wrapper_properties = TensorProperties.create_from_tensor(local_shards[0])
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_peephole.py598 inp.setType(torch._C.TensorType.create_from_tensor(torch.rand([2, 2])))
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in2271 def create_from_tensor(t: Tensor) -> TensorType: ...