Lines Matching refs:PyAffineMap

381   PyAffineMapExprList(const PyAffineMap &map, intptr_t startIndex = 0,  in PyAffineMapExprList()
404 PyAffineMap affineMap;
408 bool PyAffineMap::operator==(const PyAffineMap &other) { in operator ==()
412 py::object PyAffineMap::getCapsule() { in getCapsule()
416 PyAffineMap PyAffineMap::createFromCapsule(py::object capsule) { in createFromCapsule()
420 return PyAffineMap( in createFromCapsule()
571 [](PyAffineExpr &self, PyAffineMap &other) { in populateIRAffine()
647 py::class_<PyAffineMap>(m, "AffineMap", py::module_local()) in populateIRAffine()
649 &PyAffineMap::getCapsule) in populateIRAffine()
650 .def(MLIR_PYTHON_CAPI_FACTORY_ATTR, &PyAffineMap::createFromCapsule) in populateIRAffine()
652 [](PyAffineMap &self, PyAffineMap &other) { return self == other; }) in populateIRAffine()
653 .def("__eq__", [](PyAffineMap &self, py::object &other) { return false; }) in populateIRAffine()
655 [](PyAffineMap &self) { in populateIRAffine()
662 [](PyAffineMap &self) { in populateIRAffine()
671 [](PyAffineMap &self) { in populateIRAffine()
677 pyListToVector<PyAffineMap, MlirAffineMap>( in populateIRAffine()
686 std::vector<PyAffineMap> res; in populateIRAffine()
694 [](PyAffineMap &self) { return self.getContext().getObject(); }, in populateIRAffine()
697 "dump", [](PyAffineMap &self) { mlirAffineMapDump(self); }, in populateIRAffine()
709 return PyAffineMap(context->getRef(), map); in populateIRAffine()
719 return PyAffineMap(context->getRef(), affineMap); in populateIRAffine()
727 return PyAffineMap(context->getRef(), affineMap); in populateIRAffine()
735 return PyAffineMap(context->getRef(), affineMap); in populateIRAffine()
745 return PyAffineMap(context->getRef(), affineMap); in populateIRAffine()
760 return PyAffineMap(context->getRef(), affineMap); in populateIRAffine()
766 [](PyAffineMap &self, std::vector<intptr_t> &resultPos) { in populateIRAffine()
774 return PyAffineMap(self.getContext(), affineMap); in populateIRAffine()
779 [](PyAffineMap &self, intptr_t nResults) { in populateIRAffine()
784 return PyAffineMap(self.getContext(), affineMap); in populateIRAffine()
789 [](PyAffineMap &self, intptr_t nResults) { in populateIRAffine()
794 return PyAffineMap(self.getContext(), affineMap); in populateIRAffine()
799 [](PyAffineMap &self, PyAffineExpr &expression, in populateIRAffine()
804 return PyAffineMap(self.getContext(), affineMap); in populateIRAffine()
810 [](PyAffineMap &self) { return mlirAffineMapIsPermutation(self); }) in populateIRAffine()
812 [](PyAffineMap &self) { in populateIRAffine()
817 [](PyAffineMap &self) { return mlirAffineMapGetNumDims(self); }) in populateIRAffine()
820 [](PyAffineMap &self) { return mlirAffineMapGetNumInputs(self); }) in populateIRAffine()
823 [](PyAffineMap &self) { return mlirAffineMapGetNumSymbols(self); }) in populateIRAffine()
824 .def_property_readonly("results", [](PyAffineMap &self) { in populateIRAffine()