Home | History | Annotate | Download | only in smartreply 
      1 package(default_visibility = ["//visibility:public"])
      2 
      3 load("//tensorflow/contrib/lite:build_def.bzl", "tflite_copts", "gen_selected_ops")
      4 
      5 licenses(["notice"])  # Apache 2.0
      6 
      7 gen_selected_ops(
      8     name = "smartreply_ops",
      9     model = "@tflite_smartreply//:smartreply.tflite",
     10 )
     11 
     12 cc_library(
     13     name = "custom_ops",
     14     srcs = [
     15         "ops/extract_feature.cc",
     16         "ops/normalize.cc",
     17         "ops/predict.cc",
     18         ":smartreply_ops",
     19     ],
     20     copts = tflite_copts(),
     21     deps = [
     22         "//tensorflow/contrib/lite:framework",
     23         "//tensorflow/contrib/lite:string_util",
     24         "//tensorflow/contrib/lite/kernels:builtin_ops",
     25         "//tensorflow/contrib/lite/tools:mutable_op_resolver",
     26         "@com_google_absl//absl/strings",
     27         "@com_googlesource_code_re2//:re2",
     28         "@farmhash_archive//:farmhash",
     29     ],
     30 )
     31 
     32 cc_library(
     33     name = "predictor_lib",
     34     srcs = ["predictor.cc"],
     35     hdrs = ["predictor.h"],
     36     copts = tflite_copts(),
     37     deps = [
     38         ":custom_ops",
     39         "//tensorflow/contrib/lite:framework",
     40         "//tensorflow/contrib/lite:string_util",
     41         "//tensorflow/contrib/lite/kernels:builtin_ops",
     42         "//tensorflow/contrib/lite/tools:mutable_op_resolver",
     43         "@com_google_absl//absl/strings",
     44         "@com_googlesource_code_re2//:re2",
     45     ],
     46 )
     47 
     48 cc_test(
     49     name = "extract_feature_op_test",
     50     size = "small",
     51     srcs = ["ops/extract_feature_test.cc"],
     52     deps = [
     53         ":custom_ops",
     54         "//tensorflow/contrib/lite:framework",
     55         "//tensorflow/contrib/lite/kernels:builtin_ops",
     56         "//tensorflow/contrib/lite/kernels:test_util",
     57         "@com_google_googletest//:gtest",
     58         "@farmhash_archive//:farmhash",
     59     ],
     60 )
     61 
     62 cc_test(
     63     name = "normalize_op_test",
     64     size = "small",
     65     srcs = ["ops/normalize_test.cc"],
     66     deps = [
     67         ":custom_ops",
     68         "//tensorflow/contrib/lite:framework",
     69         "//tensorflow/contrib/lite:string_util",
     70         "//tensorflow/contrib/lite/kernels:builtin_ops",
     71         "//tensorflow/contrib/lite/kernels:test_util",
     72         "@com_google_googletest//:gtest",
     73     ],
     74 )
     75 
     76 cc_test(
     77     name = "predict_op_test",
     78     size = "small",
     79     srcs = ["ops/predict_test.cc"],
     80     deps = [
     81         ":custom_ops",
     82         "//tensorflow/contrib/lite:framework",
     83         "//tensorflow/contrib/lite:string_util",
     84         "//tensorflow/contrib/lite/kernels:builtin_ops",
     85         "//tensorflow/contrib/lite/kernels:test_util",
     86         "@com_google_googletest//:gtest",
     87     ],
     88 )
     89 
     90 filegroup(
     91     name = "all_files",
     92     srcs = glob(
     93         ["**/*"],
     94         exclude = [
     95             "**/METADATA",
     96             "**/OWNERS",
     97         ],
     98     ),
     99     visibility = ["//tensorflow:__subpackages__"],
    100 )
    101