Lines Matching defs:shape

319   PyShapedTypeComponents(nb::list shape, MlirType elementType)
320       : shape(std::move(shape)), elementType(elementType), ranked(true) {}
321   PyShapedTypeComponents(nb::list shape, MlirType elementType,
323       : shape(std::move(shape)), elementType(elementType), attribute(attribute),
327       : shape(other.shape), elementType(other.elementType),
346             [](nb::list shape, PyType &elementType) {
347               return PyShapedTypeComponents(std::move(shape), elementType);
349             nb::arg("shape"), nb::arg("element_type"),
353             [](nb::list shape, PyType &elementType, PyAttribute &attribute) {
354               return PyShapedTypeComponents(std::move(shape), elementType,
357             nb::arg("shape"), nb::arg("element_type"), nb::arg("attribute"),
369               return nb::int_(self.shape.size());
375             "shape",
380               return nb::list(self.shape);
382             "Returns the shape of the ranked shaped type components as a list "
391   nb::list shape;
414   /// Appends the shaped type components provided as unpacked shape, element
417                                     const int64_t *shape, MlirType elementType,
425         shapeList.append(shape[i]);
457       throw nb::value_error("Failed to infer result shape type components");