Searched refs:zero_dim_tensor (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/torch/jit/ |
H A D | _shape_functions.py | 940 def zero_dim_tensor(input: Any): function 1241 "prim::NumToTensor.Scalar(Scalar a) -> Tensor", zero_dim_tensor 1243 add_shape_compute_mapping("prim::NumToTensor.bool(bool a) -> Tensor", zero_dim_tensor) 1389 "aten::mean(Tensor self, *, ScalarType? dtype=None) -> Tensor", zero_dim_tensor 1392 "aten::sum(Tensor self, *, ScalarType? dtype=None) -> Tensor", zero_dim_tensor
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_view_ops.py | 1054 zero_dim_tensor = torch.tensor(123, device=device) 1055 flat0 = zero_dim_tensor.ravel() 1057 flat1 = zero_dim_tensor.ravel() 1061 self.assertEqual(zero_dim_tensor.shape, torch.Size([])) 1194 zero_dim_tensor = torch.tensor(123, device=device) 1195 flat0 = zero_dim_tensor.flatten() 1197 flat1 = zero_dim_tensor.flatten() 1199 self.assertEqual(zero_dim_tensor.shape, torch.Size([]))
|
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
H A D | _optim_utils.py | 330 for state_name, zero_dim_tensor in sorted_items(zero_dim_tensor_state): 331 unflat_state_param[state_name] = zero_dim_tensor
|