Home
last modified time | relevance | path

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.py940 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 Dtest_view_ops.py1054 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.py330 for state_name, zero_dim_tensor in sorted_items(zero_dim_tensor_state):
331 unflat_state_param[state_name] = zero_dim_tensor