experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Linalg] Update/fix documentation for linalg.indexed_generic.

Browse Source 
PiperOrigin-RevId: 283503642
Change-Id: Iee41c0a75c73e0f9b6e5359b71b1f66f814e100c
This commit is contained in:
Alexander Belyaev 2019-12-03 01:55:18 -08:00 committed by TensorFlower Gardener
parent 8fad723f04
commit a56901ffc9
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
third_party/mlir/include/mlir/Dialect/Linalg/IR/LinalgLibraryOps.td vendored
View File

@ -567,8 +567,8 @@ def IndexedGenericOp : GenericOpBase<"indexed_generic"> {
To support inplace updates in a generic fashion, the signature of the To support inplace updates in a generic fashion, the signature of the
function must be: function must be:
``` ```
fun([input views element types], [output views element types]) fun([index types for induction variables], [input views element types],
-> ([output views element types]) [output views element types]) -> ([output views element types])
``` ```
- indexing_maps: a list of AffineMapAttr, one AffineMapAttr per each input - indexing_maps: a list of AffineMapAttr, one AffineMapAttr per each input
and output view. Such AffineMapAttr specifies the mapping between the and output view. Such AffineMapAttr specifies the mapping between the
@ -587,7 +587,7 @@ def IndexedGenericOp : GenericOpBase<"indexed_generic"> {
Example: Example:
Defining a #matmul_trait attribute in MLIR can be done as follows: Defining a #matmul_trait attribute in MLIR can be done as follows:
```mlir ```mlir
func @fma(%a: f32, %b: f32, %c: f32) -> f32 { func @fma(%i: index, %j: index, %k: index, %a: f32, %b: f32, %c: f32) -> f32 {
%d = mulf %a, %b: f32 %d = mulf %a, %b: f32
%e = addf %c, %d: f32 %e = addf %c, %d: f32
return %e: f32 return %e: f32