Searched refs:item_shape (Results 1 – 3 of 3) sorted by relevance
479 shape_inference::ShapeHandle item_shape = c->input(2); in __anon9187032f0f02() local481 c->Merge(item_shape, list_shape_type.shape, &item_shape)); in __anon9187032f0f02()
1987 ArrayRef<int64_t> item_shape = data_shape.drop_front(index_rank); in Verify() local1989 inferred_item_shape = llvm::to_vector<4>(item_shape); in Verify()1993 if (failed(mlir::verifyCompatibleShape(item_shape, *inferred_item_shape))) in Verify()1997 << "] and [" << item_shape << "] don't match"; in Verify()1998 for (int i = 0, e = item_shape.size(); i < e; ++i) { in Verify()2000 int64_t dim = item_shape[i]; in Verify()
281 ArrayRef<int64_t> item_shape = out_ty.getShape().drop_front(1); in matchAndRewrite() local282 auto item_ty = RankedTensorType::get(item_shape, out_ty.getElementType()); in matchAndRewrite()286 packed_shape.append(item_shape.begin(), item_shape.end()); in matchAndRewrite()