Searched refs:new_lhs_shape (Results 1 – 2 of 2) sorted by relevance
200 Shape new_lhs_shape = lhs->shape(); in TryResolvePaddedShapesForTensorCore() local214 return std::make_tuple(&new_lhs_shape, &new_rhs_shape, in TryResolvePaddedShapesForTensorCore()218 &new_lhs_shape); in TryResolvePaddedShapesForTensorCore()220 return std::make_tuple(&new_lhs_shape, &new_result_shape, in TryResolvePaddedShapesForTensorCore()274 if (!check_size_increase(lhs->shape(), new_lhs_shape) || in TryResolvePaddedShapesForTensorCore()281 if (ShapeUtil::Equal(lhs->shape(), new_lhs_shape) && in TryResolvePaddedShapesForTensorCore()287 new_input_shapes_ptr->push_back(new_lhs_shape); in TryResolvePaddedShapesForTensorCore()
1384 llvm::SmallVector<int64_t> new_lhs_shape{N}; in matchAndRewrite() local1389 new_lhs_shape.append(lhs_shape_end.begin(), lhs_shape_end.end()); in matchAndRewrite()1395 rewriter.getI64ArrayAttr(new_lhs_shape)); in matchAndRewrite()