experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/mlir/lite
History
TensorFlower Gardener 7a6fa7dbc0 Merge pull request #41790 from lgeiger:fix-matmul-fusion 
PiperOrigin-RevId: 341026586
Change-Id: Idb578a4cf48c82abaad2f811612c4df4c2f2752c
2020-11-06 13:01:02 +00:00
..
common
Replace instances of "whitelist" with "allowlist" where possible. See Google Developer guidelines at https://developers.google.com/style/word-list#blacklist for more information.
2020-07-08 10:13:43 -07:00
experimental/estimators
Add target environment constraints to more TensorFlow packages and targets.
2020-09-24 22:38:24 -07:00
ir
[tf.lite] Add cumsum mlir conversion & e2e tests.
2020-10-19 20:16:59 -07:00
python
Update TFLite converter to export saved model signature def to the tflite schema.
2020-10-15 16:21:43 -07:00
quantization
Add target environment constraints to more TensorFlow packages and targets.
2020-09-24 22:38:24 -07:00
sparsity
Add target environment constraints to more TensorFlow packages and targets.
2020-09-24 22:38:24 -07:00
tests
Merge pull request #41790 from lgeiger:fix-matmul-fusion
2020-11-06 13:01:02 +00:00
transforms
Merge pull request #41790 from lgeiger:fix-matmul-fusion
2020-11-06 13:01:02 +00:00
utils
Merge pull request #41790 from lgeiger:fix-matmul-fusion
2020-11-06 13:01:02 +00:00
BUILD
Internal change
2020-10-13 07:16:11 -07:00
converter_gen.cc
Integrate LLVM at llvm/llvm-project@e1669843f2
2020-09-16 06:00:29 -07:00
emit_error_reporter.cc
…
emit_error_reporter.h
Change comment of the external repo #include file origin.
2020-03-23 10:42:53 -07:00
flatbuffer_export_flags.h
Make flatbuffer_translate_lib dynamic linked
2020-03-23 18:29:10 -07:00
flatbuffer_export.cc
Update TFLite converter to export saved model signature def to the tflite schema.
2020-10-15 16:21:43 -07:00
flatbuffer_export.h
Update TFLite converter to export saved model signature def to the tflite schema.
2020-10-15 16:21:43 -07:00
flatbuffer_import.cc
Refactor reading builtin code in TFLite
2020-10-05 15:36:21 -07:00
flatbuffer_import.h
Change comment of the external repo #include file origin.
2020-03-23 10:42:53 -07:00
flatbuffer_operator.cc
Integrate LLVM at llvm/llvm-project@202766947e
2020-09-02 02:32:14 -07:00
flatbuffer_operator.h
Add a TFL_CustomOp definition and support in exporter and importer.
2020-04-27 21:09:48 -07:00
flatbuffer_to_string.cc
Change comment of the external repo #include file origin.
2020-03-23 10:42:53 -07:00
flatbuffer_translate_flags.h
Make flatbuffer_translate_lib dynamic linked
2020-03-22 22:49:32 -07:00
flatbuffer_translate.cc
Integrate LLVM at llvm/llvm-project@f164534ca8
2020-08-23 13:42:08 -07:00
flatbuffer_translate.h
Change comment of the external repo #include file origin.
2020-03-23 10:42:53 -07:00
json_to_flatbuffer.cc
Create a json to flatbuffer utility for tests that cannot use mlir-flatbuffer-mlir round-trip.
2020-04-09 19:32:39 -07:00
mlir_tflite_runner.cc
Remove dependency on Dialect global registration from //tensorflow/compiler/mlir/lite/...
2020-08-24 03:22:05 -07:00
README.md
[TFLite] Update README to reflect current state
2020-03-05 13:53:04 -08:00
tf_tfl_passes.cc
- Add layout optimization passes to TFLite passes. This will change data layout for some ops (e.g. conv2d) so it will change the layout to NHWC and can be converted to tflite.
2020-09-14 18:25:23 -07:00
tf_tfl_passes.h
Add passes for saved model variable freezing in TFLite
2020-06-25 22:49:11 -07:00
tf_tfl_translate_cl.cc
Add a flag for enabling/disabling legalize tf while pass in tf_tfl_translate
2020-03-24 16:26:18 -07:00
tf_tfl_translate_cl.h
Add a flag for enabling/disabling legalize tf while pass in tf_tfl_translate
2020-03-24 16:26:18 -07:00
tf_tfl_translate.cc
Update TFLite converter to export saved model signature def to the tflite schema.
2020-10-15 16:21:43 -07:00
tf_to_tfl_flatbuffer.cc
Update TFLite converter to export saved model signature def to the tflite schema.
2020-10-15 16:21:43 -07:00
tf_to_tfl_flatbuffer.h
Update TFLite converter to export saved model signature def to the tflite schema.
2020-10-15 16:21:43 -07:00

README.md

The new MLIR based

TensorFlow to TensorFlow Lite converter

This directory contains:

  1. MLIR dialects, transformation passes and utilities for TensorFlow Lite.

API:

The API for converting TensorFlow models to TensorFlow Lite will be through tf.lite.TFLiteConverter. All the conversion code is open sourced, and the API will be integrated soon.

The conversion process from TensorFlow to TensorFlow Lite includes the

following major passes:

  • Import from GraphDef, in .pb or .pbtxt format, into MLIR.
  • Raise to Control-flow-graph. Converts TF Control Flow dialect to TF dialect.
  • The Canonicalization pass iteratively applies canonicalization transformations in a greedy way until no further changes occur. Canonicalization includes constant folding.
  • The Legalize pass converts TensorFlow operations to TensorFlow Lite ones. The operations that cannot be mapped to TensorFlow Lite dialect are left as TensorFlow operations. Unsupported op handling follows the proposed TFLite mechanism.
  • Optimizations are performed in both the TF & TFLite dialect; aiming for small size and high performance (among the core value proposition of TensorFlow Lite models).
  • The Export pass writes out TensorFlow Lite FlatBuffer format. This pass operates on MLIR TensorFlow Lite dialect and is simple/direct translation.

See https://github.com/tensorflow/tensorflow/blob/master/tensorflow/compiler/mlir/lite/tf_tfl_passes.cc for the full list of MLIR passes for conversion from TensorFlow to TensorFlow Lite.