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)
|