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");