Searched refs:complex_to_float (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
H A D | _core.py | 166 complex_to_float: bool = True, 175 _set_shape_type(value, meta_val, complex_to_float=complex_to_float) 185 complex_to_float: bool, 199 if complex_to_float: 353 _set_shape_types(outputs, node.meta["val"], complex_to_float=False) 356 _set_shape_type(outputs[0], node.meta["val"], complex_to_float=False) 483 _set_shape_types(outputs, node.meta["val"], complex_to_float=True) 488 _set_shape_type(outputs, node.meta["val"], complex_to_float=True) 525 _set_shape_type(input_, node.meta["val"], complex_to_float=lower != "none") 914 complex_to_float=lower != "none",
|
/aosp_15_r20/external/pytorch/torch/csrc/utils/ |
H A D | byte_order.cpp | 428 std::vector<T> complex_to_float(const c10::complex<T>* src, size_t len) { in complex_to_float() function 444 auto new_src = complex_to_float(src, len); in THP_encodeComplexFloatBuffer() 460 auto new_src = complex_to_float(src, len); in THP_encodeComplexDoubleBuffer()
|