Lines Matching defs:shape

134   llvm::SmallVector<int64_t> shape;
144   if (mlir::failed(parser.parseDimensionList(shape))) {
145     parser.emitError(shapeLoc, "failed to parse parameter 'shape'");
178   return TensorDescType::get(parser.getContext(), shape, elementType,
186   auto shape = getShape();
187   for (int64_t dim : shape) {
206 TensorDescType TensorDescType::get(llvm::ArrayRef<int64_t> shape,
214   return Base::get(context, shape, elementType, attr, sg_map);
217 TensorDescType TensorDescType::get(llvm::ArrayRef<int64_t> shape,
223   return Base::get(context, shape, elementType, attr, sg_map);