Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/utils/tensorboard/
H A Dsummary.py80 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),