[all-commits] [llvm/llvm-project] 310e96: [tosa][mlir] Support dynamic batch dimension for o...

[NatashaKnk via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 310e9636caeb2f3f02f3cc5bc2f180248061bbe5
      https://github.com/llvm/llvm-project/commit/310e9636caeb2f3f02f3cc5bc2f180248061bbe5
  Author: natashaknk <natashaknk at google.com>
  Date:   2022-01-12 (Wed, 12 Jan 2022)

  Changed paths:
    A mlir/include/mlir/Dialect/Tosa/Utils/CoversionUtils.h
    M mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp
    M mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp
    M mlir/lib/Dialect/Tosa/CMakeLists.txt
    A mlir/lib/Dialect/Tosa/Utils/ConversionUtils.cpp
    M mlir/test/Conversion/TosaToLinalg/tosa-to-linalg-named.mlir
    M mlir/test/Conversion/TosaToLinalg/tosa-to-linalg.mlir
    M utils/bazel/llvm-project-overlay/mlir/BUILD.bazel

  Log Message:
  -----------
  [tosa][mlir] Support dynamic batch dimension for ops where the batch dim is explicit

Dynamic batch for rescale, gather, max_pool, avg_pool, conv2D and depthwise_conv2D. Split helper functions into a separate header file.

Reviewed By: rsuderman

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