Add inputsSize to Python IR, like outputsSize

Test Plan: Used it in some notebooks. ghstack-source-id: ded2593d57c5ee6034e19f34cb1d267762ce9551 Pull Request resolved: pytorch#46779
dreiss · Oct 23, 2020 · 90a8f21 · 90a8f21
1 parent e0fd590
commit 90a8f21
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch/csrc/jit/python/python_ir.cpp b/torch/csrc/jit/python/python_ir.cpp
@@ -493,6 +493,7 @@ void initPythonIRBindings(PyObject* module_) {
           })
       .def("sourceRange", [](Node& n) { return n.sourceRange().str(); })
       .def("hasMultipleOutputs", [](Node& n) { return n.outputs().size() > 1; })
+      .def("inputsSize", [](Node& n) { return n.inputs().size(); })
       .def("outputsSize", [](Node& n) { return n.outputs().size(); })
       .NS(kind)
       .def("inputsAt", [](Node& n, size_t i) { return n.inputs().at(i); })