pytorch
820e81ba - add overload name for min/max with list input (#39614)

Commit
4 years ago
add overload name for min/max with list input (#39614) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39614 add overload name to differentiate prim::min.int(int a, int b) -> (int) prim::min.int(int[] l, int[] r) -> (int[]) Test Plan: verified op names for aten::min and aten::max are different before ``` prim::min.int(int a, int b) -> (int) prim::min.float(float a, float b) -> (float) prim::min.int_float(int a, float b) -> (float) prim::min.float_int(float a, int b) -> (float) prim::min(Scalar a, Scalar b) -> (Scalar) prim::max.int(int a, int b) -> (int) prim::max.float(float a, float b) -> (float) prim::max.int_float(int a, float b) -> (float) prim::max.float_int(float a, int b) -> (float) prim::max(Scalar a, Scalar b) -> (Scalar) prim::min.int(int[] l, int[] r) -> (int[]) prim::max.int(int[] l, int[] r) -> (int[]) prim::min.self_int(int[] self) -> (int) prim::max.self_int(int[] self) -> (int) prim::min.float(float[] l, float[] r) -> (float[]) prim::max.float(float[] l, float[] r) -> (float[]) prim::min.self_float(float[] self) -> (float) prim::max.self_float(float[] self) -> (float) prim::min.bool(bool[] l, bool[] r) -> (bool[]) prim::max.bool(bool[] l, bool[] r) -> (bool[]) prim::min.self_bool(bool[] self) -> (bool) prim::max.self_bool(bool[] self) -> (bool) ``` after ``` prim::min.int(int a, int b) -> (int) prim::min.float(float a, float b) -> (float) prim::min.int_float(int a, float b) -> (float) prim::min.float_int(float a, int b) -> (float) prim::min(Scalar a, Scalar b) -> (Scalar) prim::max.int(int a, int b) -> (int) prim::max.float(float a, float b) -> (float) prim::max.int_float(int a, float b) -> (float) prim::max.float_int(float a, int b) -> (float) prim::max(Scalar a, Scalar b) -> (Scalar) prim::min.int_list(int[] l, int[] r) -> (int[]) prim::max.int_list(int[] l, int[] r) -> (int[]) prim::min.self_int(int[] self) -> (int) prim::max.self_int(int[] self) -> (int) prim::min.float_list(float[] l, float[] r) -> (float[]) prim::max.float_list(float[] l, float[] r) -> (float[]) prim::min.self_float(float[] self) -> (float) prim::max.self_float(float[] self) -> (float) prim::min.bool_list(bool[] l, bool[] r) -> (bool[]) prim::max.bool_list(bool[] l, bool[] r) -> (bool[]) prim::min.self_bool(bool[] self) -> (bool) prim::max.self_bool(bool[] self) -> (bool) ``` Reviewed By: iseeyuan Differential Revision: D21914844 fbshipit-source-id: f1792a8c3b3ed6d1a4ba9705c4504f15e3665126
Author
Parents
Loading