projects / platform / upstream / llvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 11e9287) | patch
Add edsc::ops for pointwise, conv and dilated_conv
authorNicolas Vasilache <ntv@google.com>
Mon, 16 Dec 2019 21:32:02 +0000 (13:32 -0800)
committerA. Unique TensorFlower <gardener@tensorflow.org>
Mon, 16 Dec 2019 21:42:38 +0000 (13:42 -0800)
commit3c179b657583c4098d189a475d85f39ff230d924
treebd039a67c8c352618a3b87b720e7904ea3ebf2bftree | snapshot
parent11e92875f07261c64205c8b72038abf0d65729a0commit | diff
Add edsc::ops for pointwise, conv and dilated_conv

This CL adds more Linalg EDSC ops and tests to support building pointwise operations along with conv and dilated_conv.
This also fixes a bug in the existing linalg_matmul EDSC and beefs up the test.

The current set of ops is already enough to build an interesting, albeit simple, model used internally.

PiperOrigin-RevId: 285838012
mlir/include/mlir/Dialect/Linalg/EDSC/Builders.h diff | blob | history
mlir/include/mlir/Dialect/Linalg/EDSC/Intrinsics.h [new file with mode: 0644] blob
mlir/include/mlir/Dialect/Linalg/IR/LinalgLibraryOps.td diff | blob | history
mlir/include/mlir/EDSC/Intrinsics.h diff | blob | history
mlir/lib/Dialect/Linalg/EDSC/Builders.cpp diff | blob | history
mlir/test/EDSC/builder-api-test.cpp diff | blob | history
Domain: System / Toolchain;