1load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") 2load("//tensorflow:tensorflow.bzl", "tf_cc_test") 3 4package( 5 default_visibility = [ 6 "//tensorflow/compiler/mlir/lite:__subpackages__", 7 "//tensorflow/lite/python:__subpackages__", 8 "//tensorflow/lite/toco/python:__subpackages__", 9 ], 10 licenses = ["notice"], 11) 12 13cc_library( 14 name = "error_collector", 15 srcs = ["error_collector.cc"], 16 hdrs = ["error_collector.h"], 17 deps = [ 18 ":types_util", 19 "//tensorflow/lite/python/metrics:converter_error_data_proto_cc", 20 "@com_google_absl//absl/strings", 21 ], 22) 23 24cc_library( 25 name = "error_collector_inst", 26 srcs = ["error_collector_inst.cc"], 27 hdrs = ["error_collector_inst.h"], 28 deps = [ 29 ":error_collector", 30 ":types_util", 31 "//tensorflow/lite/python/metrics:converter_error_data_proto_cc", 32 "@com_google_absl//absl/strings", 33 "@llvm-project//mlir:IR", 34 "@llvm-project//mlir:Pass", 35 ], 36) 37 38tf_cc_test( 39 name = "error_collector_inst_test", 40 srcs = ["error_collector_inst_test.cc"], 41 data = [ 42 "testdata/strided_slice.mlir", 43 ], 44 deps = [ 45 ":error_collector_inst", 46 ":types_util", 47 "//tensorflow/compiler/mlir/tensorflow", 48 "//tensorflow/core:test", 49 "//tensorflow/core/platform:resource_loader", 50 "//tensorflow/stream_executor/lib", 51 "@com_google_googletest//:gtest_main", 52 "@llvm-project//llvm:Support", 53 "@llvm-project//mlir:FuncDialect", 54 "@llvm-project//mlir:IR", 55 "@llvm-project//mlir:Parser", 56 "@llvm-project//mlir:Pass", 57 "@llvm-project//mlir:Support", 58 ], 59) 60 61cc_library( 62 name = "types_util", 63 srcs = ["types_util.cc"], 64 hdrs = ["types_util.h"], 65 deps = [ 66 "//tensorflow/lite/python/metrics:converter_error_data_proto_cc", 67 "@llvm-project//llvm:Support", 68 "@llvm-project//mlir:IR", 69 ], 70) 71