1 licenses(["notice"]) # Apache 2.0 2 3 package(default_visibility = ["//tensorflow:internal"]) 4 5 load("//tensorflow:tensorflow.bzl", "py_test") 6 7 py_library( 8 name = "lite", 9 srcs = ["lite.py"], 10 # data = [ 11 # "//tensorflow/contrib/lite/toco/python:toco_from_protos", 12 # ], 13 srcs_version = "PY2AND3", 14 visibility = ["//visibility:public"], 15 deps = [ 16 ":op_hint", 17 "//tensorflow/contrib/lite/toco:model_flags_proto_py", 18 "//tensorflow/contrib/lite/toco:toco_flags_proto_py", 19 "//tensorflow/contrib/lite/toco/python:tensorflow_wrap_toco", 20 "//tensorflow/python:platform", 21 ], 22 ) 23 24 py_library( 25 name = "op_hint", 26 srcs = ["op_hint.py"], 27 srcs_version = "PY2AND3", 28 visibility = ["//visibility:public"], 29 deps = [ 30 "//tensorflow/contrib/framework:framework_py", 31 "//tensorflow/core:protos_all_py", 32 "//tensorflow/python:platform", 33 ], 34 ) 35 36 py_test( 37 name = "lite_test", 38 srcs = ["lite_test.py"], 39 srcs_version = "PY2AND3", 40 tags = ["no_oss"], 41 deps = [ 42 ":lite", 43 ":op_hint", 44 "//tensorflow/python:array_ops", 45 "//tensorflow/python:client_testlib", 46 "//tensorflow/python:dtypes", 47 "//tensorflow/python:platform_test", 48 "//tensorflow/python:session", 49 ], 50 ) 51 52 filegroup( 53 name = "all_files", 54 srcs = glob( 55 ["**/*"], 56 exclude = [ 57 "**/METADATA", 58 "**/OWNERS", 59 ], 60 ), 61 visibility = ["//tensorflow:__subpackages__"], 62 ) 63