Searched refs:_tensor_to_complex_val (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/tensorboard/ |
| H A D | summary.py | 80 def _tensor_to_complex_val(t: torch.Tensor) -> List[float]: function
104 torch.complex32: ("DT_COMPLEX32", "scomplex_val", _tensor_to_complex_val),
105 torch.chalf: ("DT_COMPLEX32", "scomplex_val", _tensor_to_complex_val),
106 torch.complex64: ("DT_COMPLEX64", "scomplex_val", _tensor_to_complex_val),
107 torch.cfloat: ("DT_COMPLEX64", "scomplex_val", _tensor_to_complex_val),
109 torch.complex128: ("DT_COMPLEX128", "dcomplex_val", _tensor_to_complex_val),
110 torch.cdouble: ("DT_COMPLEX128", "dcomplex_val", _tensor_to_complex_val),
|