[mlir][tosa] Shape inference for a few remaining easy cases:
authorRob Suderman <rob.suderman@gmail.com>
Wed, 4 Aug 2021 00:01:56 +0000 (17:01 -0700)
committerRob Suderman <rob.suderman@gmail.com>
Wed, 4 Aug 2021 00:20:32 +0000 (17:20 -0700)
commit143edeca6dfe5d9c006cdadb5be1ec4afd483ca3
treea301acf31bb253dcfb84bd0ba6b9acc95eb0020d
parent2d9759c7902c5cbc9a7e3ab623321d5578d51687
[mlir][tosa] Shape inference for a few remaining easy cases:

Handles shape inference for identity, cast, and rescale. These were missed
during the initialy elementwise work. This includes resize shape propagation
which includes both attribute and input type based propagation.

Reviewed By: jpienaar

Differential Revision: https://reviews.llvm.org/D105845
mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
mlir/test/Dialect/Tosa/tosa-infer-shapes.mlir