[all-commits] [llvm/llvm-project] 13bdb7: [mlir][tosa] Add tosa.conv2d as fully_connected ca...

not-jenni via All-commits all-commits at lists.llvm.org
Tue Nov 30 12:02:22 PST 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 13bdb7ab4a7acaea7144a042fe583d45fbb9b5c4
      https://github.com/llvm/llvm-project/commit/13bdb7ab4a7acaea7144a042fe583d45fbb9b5c4
  Author: not-jenni <jennik at google.com>
  Date:   2021-11-30 (Tue, 30 Nov 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
    M mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
    M mlir/test/Dialect/Tosa/canonicalize.mlir

  Log Message:
  -----------
  [mlir][tosa] Add tosa.conv2d as fully_connected canonicalization

For a 1x1 weight and stride of 1, the input/weight can be reshaped and passed into a fully connected op then reshaped back

Reviewed By: rsuderman

Differential Revision: https://reviews.llvm.org/D114757




More information about the All-commits mailing list