Fix parameter name and document option in linalg::promoteSubViews

PiperOrigin-RevId: 279086352
This commit is contained in:
Nicolas Vasilache 2019-11-07 07:55:49 -08:00 committed by A. Unique TensorFlower
parent 7af61f6bcd
commit a10d836c6d
1 changed files with 3 additions and 2 deletions

View File

@ -170,13 +170,14 @@ struct PromotionInfo {
/// 2. Take a full view on the buffer and `linalg.fill` it with zeros (use
/// float zero for now).
/// 3. Take a partial slice of the full view in step 2. and copy into it.
/// Infers statically sized buffers from subViews unless `dynamicBuffers` is
/// true.
///
/// Returns a list of PromotionInfo which hold the promoted buffer and the
/// full and partial views indexing into the buffer.
llvm::SmallVector<PromotionInfo, 8>
promoteSubViews(OpBuilder &b, Location loc, ArrayRef<Value *> subViews,
bool promoteSubViews = false,
OperationFolder *folder = nullptr);
bool dynamicBuffers = false, OperationFolder *folder = nullptr);
/// Returns all the operands of `linalgOp` that are not views.
/// Asserts that these operands are value types to allow transformations like