Home | History | Annotate | Download | only in kernels 
      1 # Description:
      2 # BigQueryReader implementation
      3 
      4 package(
      5     default_visibility = ["//visibility:private"],
      6 )
      7 
      8 licenses(["notice"])  # Apache 2.0
      9 
     10 load(
     11     "//tensorflow:tensorflow.bzl",
     12     "tf_cc_test",
     13     "tf_copts",
     14     "tf_kernel_library",
     15 )
     16 
     17 # For platform specific build config
     18 load(
     19     "//tensorflow/core:platform/default/build_config.bzl",
     20     "tf_proto_library",
     21 )
     22 
     23 tf_kernel_library(
     24     name = "bigquery_reader_ops",
     25     srcs = ["bigquery_reader_ops.cc"],
     26     visibility = ["//visibility:public"],
     27     deps = [
     28         ":bigquery_table_accessor",
     29         ":bigquery_table_partition_proto_cc",
     30         "//tensorflow/core:framework",
     31         "//tensorflow/core:lib",
     32         "//tensorflow/core:reader_base",
     33     ],
     34 )
     35 
     36 cc_library(
     37     name = "bigquery_table_accessor",
     38     srcs = ["bigquery_table_accessor.cc"],
     39     hdrs = ["bigquery_table_accessor.h"],
     40     copts = tf_copts(),
     41     linkstatic = 1,
     42     deps = [
     43         ":bigquery_table_partition_proto_cc",
     44         "//tensorflow/core:lib",
     45         "//tensorflow/core:protos_all_cc",
     46         "//tensorflow/core/platform/cloud:curl_http_request",
     47         "//tensorflow/core/platform/cloud:google_auth_provider",
     48     ],
     49     alwayslink = 1,
     50 )
     51 
     52 tf_cc_test(
     53     name = "bigquery_table_accessor_test",
     54     size = "small",
     55     srcs = [
     56         "bigquery_table_accessor_test.cc",
     57         "bigquery_table_accessor_test_data.h",
     58     ],
     59     deps = [
     60         ":bigquery_table_accessor",
     61         "//tensorflow/core:lib",
     62         "//tensorflow/core:lib_internal",
     63         "//tensorflow/core:protos_all_cc",
     64         "//tensorflow/core:test",
     65         "//tensorflow/core:test_main",
     66         "//tensorflow/core/platform/cloud:http_request_fake",
     67     ],
     68 )
     69 
     70 tf_proto_library(
     71     name = "bigquery_table_partition_proto",
     72     srcs = ["bigquery_table_partition.proto"],
     73     cc_api_version = 2,
     74 )
     75 
     76 tf_kernel_library(
     77     name = "gcs_config_ops",
     78     srcs = ["gcs_config_ops.cc"],
     79     visibility = ["//tensorflow:internal"],
     80     deps = [
     81         "//tensorflow/core:framework",
     82         "//tensorflow/core:lib",
     83         "//tensorflow/core/platform/cloud:curl_http_request",
     84         "//tensorflow/core/platform/cloud:gcs_file_system",
     85         "//tensorflow/core/platform/cloud:oauth_client",
     86         "@jsoncpp_git//:jsoncpp",
     87     ],
     88 )
     89