Searched refs:transform_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/checkpoint/ |
| H A D | default_planner.py | 257 return self.transform_tensor(read_item, tensor)
266 def transform_tensor(self, read_item: ReadItem, tensor: torch.Tensor): member in DefaultLoadPlanner
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 7552 transform_tensor = torch.from_numpy(rotation_ary).to(device, torch.float32)
7553 transform_tensor = transform_tensor[:2].unsqueeze(0)
7555 return transform_tensor, transform_ary, grid_ary
7625 transform_tensor = torch.from_numpy(torchRotation_ary).to(device, torch.float32)
7626 transform_tensor = transform_tensor[:3].unsqueeze(0)
7628 return transform_tensor, transform_ary, grid_ary
7935 transform_tensor, transform_ary, offset = \
7948 transform_tensor,
7978 transform_tensor, transform_ary, offset = \
7998 transform_tensor,
[all …]
|