Augment `stablehlo-legalize-quant-to-math` to support NCHW layout.
PiperOrigin-RevId: 726996529
Showing
- tensorflow/compiler/mlir/lite/ir/tfl_ops.td 4 additions, 1 deletiontensorflow/compiler/mlir/lite/ir/tfl_ops.td
- tensorflow/lite/experimental/litert/cc/BUILD 10 additions, 0 deletionstensorflow/lite/experimental/litert/cc/BUILD
- tensorflow/lite/experimental/litert/cc/litert_tensor_buffer_utils.cc 44 additions, 0 deletions...lite/experimental/litert/cc/litert_tensor_buffer_utils.cc
- tensorflow/lite/experimental/litert/cc/litert_tensor_buffer_utils.h 24 additions, 0 deletions.../lite/experimental/litert/cc/litert_tensor_buffer_utils.h
- tensorflow/lite/experimental/litert/runtime/BUILD 43 additions, 0 deletionstensorflow/lite/experimental/litert/runtime/BUILD
- tensorflow/lite/experimental/litert/runtime/gl_buffer.cc 4 additions, 0 deletionstensorflow/lite/experimental/litert/runtime/gl_buffer.cc
- tensorflow/lite/experimental/litert/runtime/gl_buffer_test.cc 2 additions, 0 deletions...orflow/lite/experimental/litert/runtime/gl_buffer_test.cc
- tensorflow/lite/experimental/litert/runtime/tensor_buffer.h 1 addition, 0 deletionstensorflow/lite/experimental/litert/runtime/tensor_buffer.h
- tensorflow/lite/experimental/litert/runtime/tensor_buffer_conversion.cc 96 additions, 0 deletions...e/experimental/litert/runtime/tensor_buffer_conversion.cc
- tensorflow/lite/experimental/litert/runtime/tensor_buffer_conversion.h 35 additions, 0 deletions...te/experimental/litert/runtime/tensor_buffer_conversion.h
- tensorflow/lite/experimental/litert/runtime/tensor_buffer_conversion_test.cc 70 additions, 0 deletions...erimental/litert/runtime/tensor_buffer_conversion_test.cc
- third_party/stablehlo/temporary.patch 286 additions, 0 deletionsthird_party/stablehlo/temporary.patch
- third_party/xla/.github/workflows/cpu_benchmarks.yml 6 additions, 13 deletionsthird_party/xla/.github/workflows/cpu_benchmarks.yml
- third_party/xla/.github/workflows/gpu_benchmarks.yml 76 additions, 0 deletionsthird_party/xla/.github/workflows/gpu_benchmarks.yml
- third_party/xla/third_party/stablehlo/temporary.patch 286 additions, 0 deletionsthird_party/xla/third_party/stablehlo/temporary.patch
Please register or sign in to comment