1package( 2 default_visibility = [ 3 "//visibility:public", 4 ], 5 features = ["-parse_headers"], 6 licenses = ["notice"], 7) 8 9cc_binary( 10 name = "option_writer_generator", 11 srcs = ["option_writer_generator.cc"], 12 deps = [ 13 "//tensorflow/lite/schema:schema_fbs_with_reflection", 14 "@flatbuffers", 15 ], 16) 17 18cc_library( 19 name = "writer_lib", 20 srcs = [ 21 "enum_mapping.h", 22 "writer_lib.cc", 23 ], 24 hdrs = [ 25 "writer_lib.h", 26 ], 27 data = [ 28 ":option_writer_gen", 29 ], 30 textual_hdrs = ["option_writer_generated.h"], 31 deps = [ 32 "//tensorflow/lite:builtin_op_data", 33 "//tensorflow/lite:framework", 34 "//tensorflow/lite:schema_fbs_version", 35 "//tensorflow/lite/c:common", 36 "//tensorflow/lite/schema:schema_conversion_utils", 37 "//tensorflow/lite/schema:schema_fbs_with_reflection", 38 "//tensorflow/lite/tools/versioning", 39 "@com_google_absl//absl/container:flat_hash_map", 40 "@com_google_absl//absl/container:flat_hash_set", 41 ], 42) 43 44cc_binary( 45 name = "writer", 46 srcs = ["writer.cc"], 47 deps = [ 48 ":writer_lib", 49 "//tensorflow/lite:framework", 50 "//tensorflow/lite/kernels:builtin_ops", 51 ], 52) 53 54cc_binary( 55 name = "writer_test", 56 srcs = ["writer_test.cc"], 57 deps = [ 58 ":writer_lib", 59 "//tensorflow/lite:framework", 60 "//tensorflow/lite/kernels:builtin_ops", 61 ], 62) 63 64cc_test( 65 name = "writer_lib_test", 66 size = "small", 67 srcs = ["writer_lib_test.cc"], 68 deps = [ 69 ":writer_lib", 70 "//tensorflow/lite:framework", 71 "//tensorflow/lite/c:c_api_types", 72 "//tensorflow/lite/c:common", 73 "//tensorflow/lite/kernels:builtin_ops", 74 "//tensorflow/lite/kernels:subgraph_test_util", 75 "//tensorflow/lite/schema:schema_fbs", 76 "//tensorflow/lite/testing:util", 77 "@com_google_googletest//:gtest_main", 78 ], 79) 80 81genrule( 82 name = "option_writer_gen", 83 outs = ["option_writer_generated.h"], 84 cmd = "$(location :option_writer_generator) $(@)", 85 tools = [":option_writer_generator"], 86) 87