Lines Matching refs:PyAffineExpr

76 template <typename DerivedTy, typename BaseTy = PyAffineExpr>
89 PyConcreteAffineExpr(PyAffineExpr &orig) in PyConcreteAffineExpr()
92 static MlirAffineExpr castFrom(PyAffineExpr &orig) { in castFrom()
104 cls.def(py::init<PyAffineExpr &>(), py::arg("expr")); in bind()
107 [](PyAffineExpr &otherAffineExpr) -> bool { in bind()
186 PyAffineExpr lhs() { in lhs()
188 return PyAffineExpr(getContext(), lhsExpr); in lhs()
191 PyAffineExpr rhs() { in rhs()
193 return PyAffineExpr(getContext(), rhsExpr); in rhs()
209 static PyAffineAddExpr get(PyAffineExpr lhs, const PyAffineExpr &rhs) { in get()
214 static PyAffineAddExpr getRHSConstant(PyAffineExpr lhs, intptr_t rhs) { in getRHSConstant()
220 static PyAffineAddExpr getLHSConstant(intptr_t lhs, PyAffineExpr rhs) { in getLHSConstant()
238 static PyAffineMulExpr get(PyAffineExpr lhs, const PyAffineExpr &rhs) { in get()
243 static PyAffineMulExpr getRHSConstant(PyAffineExpr lhs, intptr_t rhs) { in getRHSConstant()
249 static PyAffineMulExpr getLHSConstant(intptr_t lhs, PyAffineExpr rhs) { in getLHSConstant()
267 static PyAffineModExpr get(PyAffineExpr lhs, const PyAffineExpr &rhs) { in get()
272 static PyAffineModExpr getRHSConstant(PyAffineExpr lhs, intptr_t rhs) { in getRHSConstant()
278 static PyAffineModExpr getLHSConstant(intptr_t lhs, PyAffineExpr rhs) { in getLHSConstant()
296 static PyAffineFloorDivExpr get(PyAffineExpr lhs, const PyAffineExpr &rhs) { in get()
301 static PyAffineFloorDivExpr getRHSConstant(PyAffineExpr lhs, intptr_t rhs) { in getRHSConstant()
307 static PyAffineFloorDivExpr getLHSConstant(intptr_t lhs, PyAffineExpr rhs) { in getLHSConstant()
325 static PyAffineCeilDivExpr get(PyAffineExpr lhs, const PyAffineExpr &rhs) { in get()
330 static PyAffineCeilDivExpr getRHSConstant(PyAffineExpr lhs, intptr_t rhs) { in getRHSConstant()
336 static PyAffineCeilDivExpr getLHSConstant(intptr_t lhs, PyAffineExpr rhs) { in getLHSConstant()
349 bool PyAffineExpr::operator==(const PyAffineExpr &other) { in operator ==()
353 py::object PyAffineExpr::getCapsule() { in getCapsule()
358 PyAffineExpr PyAffineExpr::createFromCapsule(py::object capsule) { in createFromCapsule()
362 return PyAffineExpr( in createFromCapsule()
377 : public Sliceable<PyAffineMapExprList, PyAffineExpr> {
390 friend class Sliceable<PyAffineMapExprList, PyAffineExpr>;
394 PyAffineExpr getRawElement(intptr_t pos) { in getRawElement()
395 return PyAffineExpr(affineMap.getContext(), in getRawElement()
435 PyAffineExpr getExpr() { in getExpr()
436 return PyAffineExpr(set.getContext(), in getExpr()
507 py::class_<PyAffineExpr>(m, "AffineExpr", py::module_local()) in populateIRAffine()
509 &PyAffineExpr::getCapsule) in populateIRAffine()
510 .def(MLIR_PYTHON_CAPI_FACTORY_ATTR, &PyAffineExpr::createFromCapsule) in populateIRAffine()
520 [](PyAffineExpr &self, intptr_t other) { in populateIRAffine()
526 [](PyAffineExpr &self, PyAffineExpr &other) { in populateIRAffine()
533 [](PyAffineExpr &self, intptr_t other) { in populateIRAffine()
539 [](PyAffineExpr &self, intptr_t other) { in populateIRAffine()
543 .def("__eq__", [](PyAffineExpr &self, in populateIRAffine()
544 PyAffineExpr &other) { return self == other; }) in populateIRAffine()
546 [](PyAffineExpr &self, py::object &other) { return false; }) in populateIRAffine()
548 [](PyAffineExpr &self) { in populateIRAffine()
555 [](PyAffineExpr &self) { in populateIRAffine()
564 [](PyAffineExpr &self) { in populateIRAffine()
569 [](PyAffineExpr &self) { return self.getContext().getObject(); }) in populateIRAffine()
571 [](PyAffineExpr &self, PyAffineMap &other) { in populateIRAffine()
572 return PyAffineExpr(self.getContext(), in populateIRAffine()
632 "dump", [](PyAffineExpr &self) { mlirAffineExprDump(self); }, in populateIRAffine()
704 pyListToVector<PyAffineExpr, MlirAffineExpr>( in populateIRAffine()
799 [](PyAffineMap &self, PyAffineExpr &expression, in populateIRAffine()
800 PyAffineExpr &replacement, intptr_t numResultDims, in populateIRAffine()
882 pyListToVector<PyAffineExpr>(exprs, affineExprs, in populateIRAffine()
917 pyListToVector<PyAffineExpr>( in populateIRAffine()
920 pyListToVector<PyAffineExpr>( in populateIRAffine()