Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset11.py107 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 Dsymbolic_opset9.py1218 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 Dsymbolic_helper.py1648 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)