|
|
@ -27,7 +27,9 @@ atan2_op_info = TBERegOp("Atan2") \
|
|
|
|
.input(0, "x1", False, "required", "all") \
|
|
|
|
.input(0, "x1", False, "required", "all") \
|
|
|
|
.input(1, "x2", False, "required", "all") \
|
|
|
|
.input(1, "x2", False, "required", "all") \
|
|
|
|
.output(0, "y", False, "required", "all") \
|
|
|
|
.output(0, "y", False, "required", "all") \
|
|
|
|
|
|
|
|
.dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
|
|
|
|
.dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
|
|
|
|
.dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
|
|
|
|
|
|
|
|
.dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
|
|
|
|
.dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
|
|
|
|
.dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
|
|
|
|
.get_op_info()
|
|
|
|
.get_op_info()
|
|
|
|
|
|
|
|
|
|
|
|