projects / platform / upstream / llvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 03cdb52) | patch
[mlir][tosa] Added shape inference for tosa convolution operations
authorRob Suderman <rob.suderman@gmail.com>
Mon, 19 Jul 2021 17:31:02 +0000 (10:31 -0700)
committerRob Suderman <rob.suderman@gmail.com>
Mon, 19 Jul 2021 17:41:56 +0000 (10:41 -0700)
commit11dda1a234620639a856de61f8a2c0dc3a8bdd2a
tree628ff730706158f254f2b7cbd7e6884065334b21tree | snapshot
parent03cdb5221d120c4000725e4aa862ef8c8b852274commit | diff
[mlir][tosa] Added shape inference for tosa convolution operations

Added shape inference handles cases for convolution operations. This includes
conv2d, conv3d, depthwise_conv2d, and transpose_conv2d. With transpose conv
we use the specified output shape when possible however will shape propagate
if the output shape attribute has dynamic values.

Reviewed By: jpienaar

Differential Revision: https://reviews.llvm.org/D105645
mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td diff | blob | history
mlir/lib/Dialect/Tosa/IR/TosaOps.cpp diff | blob | history
mlir/test/Dialect/Tosa/tosa-infer-shapes.mlir diff | blob | history
Domain: System / Toolchain;