Searched refs:new_shape_array (Results 1 – 3 of 3) sorted by relevance
73 llvm::ArrayRef<int64_t> new_shape_array, in InsertReshapeOp() argument76 new_shape_array.size(), builder->getIntegerType(32)); in InsertReshapeOp()81 for (auto size : new_shape_array) { in InsertReshapeOp()89 auto reshape_out_type = RankedTensorType::get(new_shape_array, element_type); in InsertReshapeOp()
1494 auto new_shape_array = output_type.getShape(); in matchAndRewrite() local1497 for (auto size : new_shape_array) { in matchAndRewrite()
1982 SmallVector<int32_t, 4> new_shape_array; in matchAndRewrite() local1984 new_shape_array.push_back(static_cast<int32_t>(size)); in matchAndRewrite()1989 RankedTensorType::get(new_shape_array.size(), in matchAndRewrite()1991 new_shape_array)); in matchAndRewrite()