Searched refs:_op_with_optional_float_cast (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/torch/onnx/ |
H A D | symbolic_opset11.py | 107 return symbolic_helper._op_with_optional_float_cast( 140 return symbolic_helper._op_with_optional_float_cast( 153 return symbolic_helper._op_with_optional_float_cast( 157 return symbolic_helper._op_with_optional_float_cast( 168 return symbolic_helper._op_with_optional_float_cast( 172 return symbolic_helper._op_with_optional_float_cast(
|
H A D | symbolic_opset9.py | 1218 return symbolic_helper._op_with_optional_float_cast( 1564 input = symbolic_helper._op_with_optional_float_cast( 1758 return symbolic_helper._op_with_optional_float_cast( 1811 return symbolic_helper._op_with_optional_float_cast( 1823 return symbolic_helper._op_with_optional_float_cast( 3125 return symbolic_helper._op_with_optional_float_cast( 3141 return symbolic_helper._op_with_optional_float_cast( 3147 return symbolic_helper._op_with_optional_float_cast( 3156 return symbolic_helper._op_with_optional_float_cast( 3162 return symbolic_helper._op_with_optional_float_cast( [all …]
|
H A D | symbolic_helper.py | 1648 def _op_with_optional_float_cast(g: jit_utils.GraphContext, op_name, *args, **kwargs): function 1828 return _op_with_optional_float_cast(g, "Max", self, dim_or_y, opset_before=12) 1848 return _op_with_optional_float_cast(g, "Min", self, dim_or_y, opset_before=12)
|