| load("//tensorflow/compiler/mlir:glob_lit_test.bzl", "glob_lit_tests") |
| load("//tensorflow:tensorflow.bzl", "tf_native_cc_binary") |
| |
| licenses(["notice"]) |
| |
| glob_lit_tests( |
| data = [ |
| ":extra_files", |
| ":test_utilities", |
| ], |
| driver = "@llvm-project//mlir:run_lit.sh", |
| test_file_exts = [ |
| "mlir", |
| "cc", |
| "json", |
| ], |
| ) |
| |
| # Bundle together all of the test utilities that are used by tests. |
| filegroup( |
| name = "test_utilities", |
| testonly = True, |
| data = [ |
| ":importer_test_legacy_reshape", |
| ":importer_test_min_max", |
| ":schema.fbs", |
| "//tensorflow/compiler/mlir/lite:flatbuffer_to_string", |
| "//tensorflow/compiler/mlir/lite:flatbuffer_translate", |
| "//tensorflow/compiler/mlir/lite:json_to_flatbuffer", |
| "@llvm-project//llvm:FileCheck", |
| ], |
| ) |
| |
| # Bundle together all the files that are used by the non-mlir file-based tests. |
| filegroup( |
| name = "extra_files", |
| srcs = glob( |
| [ |
| "**/importer_test_min_max.cc.mlir", |
| "**/reshape.mlir", |
| ], |
| ), |
| ) |
| |
| # A binary to inject min/max to a tflite model. |
| # A file check command is used to verify the imported result from this |
| # binary format. |
| tf_native_cc_binary( |
| name = "importer_test_min_max", |
| srcs = [ |
| "importer_test_min_max.cc", |
| ], |
| deps = [ |
| "//tensorflow/lite:framework", |
| "//tensorflow/lite/schema:schema_fbs", |
| "@com_google_absl//absl/strings", |
| "@llvm-project//llvm:support", |
| ], |
| ) |
| |
| # A binary to produce legacy shape op which doesn't use the second operand. |
| # A file check command is used to verify the imported result from this |
| # binary format. |
| tf_native_cc_binary( |
| name = "importer_test_legacy_reshape", |
| srcs = [ |
| "importer_test_legacy_reshape.cc", |
| ], |
| deps = [ |
| "//tensorflow/lite:framework", |
| "//tensorflow/lite/schema:schema_fbs", |
| "@com_google_absl//absl/strings", |
| "@llvm-project//llvm:support", |
| ], |
| ) |