Home
last modified time | relevance | path

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.py166 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 Dbyte_order.cpp428 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()