|
|
|
@ -157,6 +157,7 @@ tf_kernel_library(
|
|
|
|
|
name = "collective_ops",
|
|
|
|
|
prefix = "collective_ops",
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:collective_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
@ -221,6 +222,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
],
|
|
|
|
|
alwayslink = 1,
|
|
|
|
@ -311,6 +313,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core/nccl:nccl_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:gpu_headers_lib",
|
|
|
|
|
"//tensorflow/core:nccl_ops_op_lib",
|
|
|
|
|
]),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -456,6 +459,7 @@ cc_library(
|
|
|
|
|
name = "batch_kernels",
|
|
|
|
|
srcs = ["batch_kernels.cc"],
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:batch_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework_headers_lib",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
|
"//tensorflow/core/kernels:concat_lib_hdrs",
|
|
|
|
@ -682,6 +686,7 @@ ARRAY_DEPS = [
|
|
|
|
|
":ops_util",
|
|
|
|
|
":transpose_functor",
|
|
|
|
|
"//tensorflow/core:array_grad",
|
|
|
|
|
"//tensorflow/core:array_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:core_cpu",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
@ -710,6 +715,7 @@ tf_kernel_library(
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:framework_headers_lib",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:set_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
@ -1073,6 +1079,7 @@ tf_kernel_library(
|
|
|
|
|
srcs = ["ragged_gather_op.cc"],
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:ragged_array_ops_op_lib",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -1083,6 +1090,7 @@ tf_cc_test(
|
|
|
|
|
deps = [
|
|
|
|
|
":ragged_gather_op",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:ragged_array_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:test",
|
|
|
|
|
"//tensorflow/core:test_main",
|
|
|
|
|
"//tensorflow/core:testlib",
|
|
|
|
@ -1095,6 +1103,7 @@ tf_kernel_library(
|
|
|
|
|
srcs = ["ragged_range_op.cc"],
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:ragged_math_ops_op_lib",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -1104,6 +1113,7 @@ tf_cc_test(
|
|
|
|
|
deps = [
|
|
|
|
|
":ragged_range_op",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:ragged_math_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:test",
|
|
|
|
|
"//tensorflow/core:test_main",
|
|
|
|
|
"//tensorflow/core:testlib",
|
|
|
|
@ -1116,6 +1126,7 @@ tf_kernel_library(
|
|
|
|
|
srcs = ["ragged_tensor_to_sparse_kernel.cc"],
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:ragged_conversion_ops_op_lib",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -1127,6 +1138,7 @@ tf_cc_test(
|
|
|
|
|
":ragged_tensor_to_sparse_kernel",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:ragged_conversion_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:test",
|
|
|
|
|
"//tensorflow/core:test_main",
|
|
|
|
|
"//tensorflow/core:testlib",
|
|
|
|
@ -1140,6 +1152,7 @@ tf_kernel_library(
|
|
|
|
|
visibility = ["//visibility:public"],
|
|
|
|
|
deps = [
|
|
|
|
|
":gpu_util_hdrs",
|
|
|
|
|
"//tensorflow/core:cudnn_rnn_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
@ -1216,6 +1229,7 @@ tf_cuda_cc_test(
|
|
|
|
|
"//tensorflow/core:core_cpu",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:math_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
|
"//tensorflow/core:test",
|
|
|
|
|
"//tensorflow/core:test_main",
|
|
|
|
@ -1740,6 +1754,7 @@ tf_kernel_library(
|
|
|
|
|
prefix = "candidate_sampler_ops",
|
|
|
|
|
deps = [
|
|
|
|
|
":range_sampler",
|
|
|
|
|
"//tensorflow/core:candidate_sampling_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
],
|
|
|
|
@ -1773,6 +1788,7 @@ tf_kernel_library(
|
|
|
|
|
name = "control_flow_ops",
|
|
|
|
|
prefix = "control_flow_ops",
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:control_flow_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
],
|
|
|
|
@ -1784,6 +1800,7 @@ tf_kernel_library(
|
|
|
|
|
deps = [
|
|
|
|
|
":bounds_check",
|
|
|
|
|
":ops_util",
|
|
|
|
|
"//tensorflow/core:ctc_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core/util/ctc:ctc_beam_search_lib",
|
|
|
|
@ -1864,6 +1881,7 @@ DATA_FLOW_DEPS = [
|
|
|
|
|
":typed_queue",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
"//tensorflow/core:core_cpu",
|
|
|
|
|
"//tensorflow/core:data_flow_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
@ -1928,6 +1946,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:scoped_allocator_ops_op_lib",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -1946,6 +1965,7 @@ tf_cuda_cc_test(
|
|
|
|
|
"//tensorflow/core:core_cpu_internal",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:math_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:proto_text",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
|
"//tensorflow/core:test",
|
|
|
|
@ -1993,6 +2013,7 @@ tf_kernel_library(
|
|
|
|
|
DYNAMIC_DEPS = [
|
|
|
|
|
":bounds_check",
|
|
|
|
|
"//tensorflow/core:core_cpu",
|
|
|
|
|
"//tensorflow/core:data_flow_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
@ -2025,6 +2046,7 @@ LOOKUP_DEPS = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:lookup_ops_op_lib",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
tf_kernel_library(
|
|
|
|
@ -2053,6 +2075,7 @@ tf_kernel_library(
|
|
|
|
|
deps = [
|
|
|
|
|
":lookup_table_init_op",
|
|
|
|
|
":lookup_table_op",
|
|
|
|
|
"//tensorflow/core:checkpoint_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
@ -2063,6 +2086,7 @@ tf_kernel_library(
|
|
|
|
|
name = "load_and_remap_matrix_op",
|
|
|
|
|
srcs = ["load_and_remap_matrix_op.cc"],
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:checkpoint_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
@ -2209,6 +2233,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:resource_variable_ops_op_lib",
|
|
|
|
|
"@com_google_absl//absl/strings",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
@ -2225,6 +2250,7 @@ tf_kernel_library(
|
|
|
|
|
":concat_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:list_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
@ -2235,6 +2261,7 @@ tf_kernel_library(
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:user_ops_op_lib",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -2257,6 +2284,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:core_cpu",
|
|
|
|
|
"//tensorflow/core:core_cpu_internal",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:functional_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
@ -2269,6 +2297,7 @@ tf_kernel_library(
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:core_cpu_internal",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:functional_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
|
"//tensorflow/core/grappler:grappler_item",
|
|
|
|
@ -2311,6 +2340,7 @@ IMAGE_DEPS = [
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:gif_internal",
|
|
|
|
|
"//tensorflow/core:image_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:jpeg_internal",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
@ -2650,6 +2680,7 @@ cc_library(
|
|
|
|
|
IO_DEPS = [
|
|
|
|
|
":ops_util",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:io_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
@ -2693,6 +2724,7 @@ SAVE_RESTORE_DEPS = [
|
|
|
|
|
":bounds_check_lib",
|
|
|
|
|
":save_restore_tensor",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:io_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
@ -2811,6 +2843,7 @@ LINALG_DEPS = [
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:linalg_ops_op_lib",
|
|
|
|
|
] + if_cuda([
|
|
|
|
|
":cuda_solvers",
|
|
|
|
|
":transpose_functor",
|
|
|
|
@ -2954,6 +2987,7 @@ LOGGING_DEPS = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:logging_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
@ -3025,6 +3059,7 @@ tf_kernel_library(
|
|
|
|
|
":bounds_check",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:manip_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
@ -3056,6 +3091,7 @@ MATH_DEPS = [
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:math_grad",
|
|
|
|
|
"//tensorflow/core:math_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
@ -3166,7 +3202,7 @@ tf_kernel_library(
|
|
|
|
|
tf_kernel_library(
|
|
|
|
|
name = "cwise_op",
|
|
|
|
|
prefix = "cwise_op",
|
|
|
|
|
deps = MATH_DEPS,
|
|
|
|
|
deps = MATH_DEPS + ["//tensorflow/core:bitwise_ops_op_lib"],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
tf_kernel_library(
|
|
|
|
@ -3185,6 +3221,7 @@ tf_kernel_library(
|
|
|
|
|
name = "fft_ops",
|
|
|
|
|
prefix = "fft_ops",
|
|
|
|
|
deps = MATH_DEPS + [
|
|
|
|
|
"//tensorflow/core:spectral_ops_op_lib",
|
|
|
|
|
] + if_cuda([
|
|
|
|
|
"//tensorflow/core/platform/default/build_config:cufft_plugin",
|
|
|
|
|
]),
|
|
|
|
@ -3383,7 +3420,10 @@ tf_cuda_cc_test(
|
|
|
|
|
":quantized_ops",
|
|
|
|
|
"//tensorflow/cc:cc_ops",
|
|
|
|
|
"//tensorflow/cc:client_session",
|
|
|
|
|
"//tensorflow/core:array_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:math_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
|
"//tensorflow/core:test",
|
|
|
|
|
"//tensorflow/core:test_main",
|
|
|
|
@ -3641,6 +3681,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
] + select({
|
|
|
|
|
":xsmm_convolutions": [
|
|
|
|
|
"@libxsmm_archive//:xsmm_avx",
|
|
|
|
@ -3662,6 +3703,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:core_cpu",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
] + if_cuda([
|
|
|
|
|
"@cub_archive//:cub",
|
|
|
|
|
"@local_config_cuda//cuda:cudnn_header",
|
|
|
|
@ -3681,6 +3723,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:core_cpu",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
] + if_cuda([
|
|
|
|
|
"@local_config_cuda//cuda:cudnn_header",
|
|
|
|
|
]),
|
|
|
|
@ -3728,8 +3771,9 @@ NN_DEPS = [
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:nn_grad",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
]
|
|
|
|
|
] + if_mkl(["//tensorflow/core:mkl_nn_ops_op_lib"])
|
|
|
|
|
|
|
|
|
|
tf_kernel_library(
|
|
|
|
|
name = "batch_norm_op",
|
|
|
|
@ -3849,6 +3893,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:nn_grad",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
] + if_cuda(["@cub_archive//:cub"]),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -3956,6 +4001,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:stream_executor",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
],
|
|
|
|
@ -3999,6 +4045,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:core_cpu",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
@ -4080,6 +4127,7 @@ cc_library(
|
|
|
|
|
PARSING_DEPS = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:parsing_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:proto_text",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
|
]
|
|
|
|
@ -4148,6 +4196,7 @@ RANDOM_OPS_DEPS = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:random_ops_op_lib",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
tf_kernel_library(
|
|
|
|
@ -4186,6 +4235,7 @@ tf_kernel_library(
|
|
|
|
|
":random_op",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:stateless_random_ops_op_lib",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -4200,6 +4250,8 @@ cc_library(
|
|
|
|
|
REQUIRED_DEPS = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:no_op_op_lib",
|
|
|
|
|
"//tensorflow/core:sendrecv_ops_op_lib",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
tf_kernel_library(
|
|
|
|
@ -4260,6 +4312,7 @@ cc_library(
|
|
|
|
|
SPARSE_DEPS = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:sparse_ops_op_lib",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
tf_kernel_library(
|
|
|
|
@ -4504,6 +4557,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:sdca_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
"@farmhash_archive//:farmhash",
|
|
|
|
|
],
|
|
|
|
@ -4543,6 +4597,7 @@ STATE_DEPS = [
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:state_ops_op_lib",
|
|
|
|
|
] + if_sycl(["//tensorflow/core:sycl_runtime"])
|
|
|
|
|
|
|
|
|
|
tf_kernel_library(
|
|
|
|
@ -4680,6 +4735,7 @@ STRING_DEPS = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:string_ops_op_lib",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
tf_kernel_library(
|
|
|
|
@ -4830,6 +4886,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:string_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
"//third_party/icu/data:conversion_data",
|
|
|
|
|
"@icu//:common",
|
|
|
|
@ -4851,6 +4908,7 @@ tf_kernel_library(
|
|
|
|
|
":variable_ops",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:training_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
@ -4909,6 +4967,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:random_ops_op_lib",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -4936,6 +4995,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:random_ops_op_lib",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -5844,9 +5904,12 @@ tf_kernel_library(
|
|
|
|
|
":ops_util",
|
|
|
|
|
":pooling_ops",
|
|
|
|
|
":quantization_utils",
|
|
|
|
|
"//tensorflow/core:array_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:core_cpu",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:math_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
"@gemmlowp",
|
|
|
|
|
],
|
|
|
|
@ -6416,6 +6479,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
|
"//tensorflow/core:remote_fused_graph_ops_op_lib",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -6570,6 +6634,8 @@ tf_mkl_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:mkl_nn_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
] + mkl_deps(),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -6619,6 +6685,8 @@ tf_mkl_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:mkl_nn_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
] + mkl_deps(),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -6637,6 +6705,8 @@ tf_mkl_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:mkl_nn_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
] + mkl_deps(),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -6650,6 +6720,8 @@ tf_mkl_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:mkl_nn_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
] + mkl_deps(),
|
|
|
|
|
)
|
|
|
|
@ -6664,6 +6736,8 @@ tf_mkl_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:mkl_nn_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:nn_ops_op_lib",
|
|
|
|
|
"//third_party/eigen3",
|
|
|
|
|
] + mkl_deps(),
|
|
|
|
|
)
|
|
|
|
@ -6804,6 +6878,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:protos_all_cc",
|
|
|
|
|
"//tensorflow/core:summary_ops_op_lib",
|
|
|
|
|
"//tensorflow/core/lib/db:sqlite",
|
|
|
|
|
] + if_not_v2([
|
|
|
|
|
"//tensorflow/contrib/tensorboard/db:schema",
|
|
|
|
@ -6818,6 +6893,7 @@ tf_kernel_library(
|
|
|
|
|
"decode_proto_op.cc",
|
|
|
|
|
],
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:decode_proto_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core/util/proto:decode",
|
|
|
|
@ -6831,6 +6907,7 @@ tf_kernel_library(
|
|
|
|
|
name = "encode_proto_op",
|
|
|
|
|
srcs = ["encode_proto_op.cc"],
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:encode_proto_ops_op_lib",
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core/util/proto:descriptors",
|
|
|
|
@ -6848,6 +6925,7 @@ tf_kernel_library(
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:lib",
|
|
|
|
|
"//tensorflow/core:lib_internal",
|
|
|
|
|
"//tensorflow/core:rpc_ops_op_lib",
|
|
|
|
|
"//tensorflow/core/util/rpc:call_container",
|
|
|
|
|
"//tensorflow/core/util/rpc:rpc_factory",
|
|
|
|
|
"//tensorflow/core/util/rpc:rpc_factory_registry",
|
|
|
|
@ -6860,6 +6938,7 @@ tf_kernel_library(
|
|
|
|
|
srcs = ["unicode_script_op.cc"],
|
|
|
|
|
deps = [
|
|
|
|
|
"//tensorflow/core:framework",
|
|
|
|
|
"//tensorflow/core:string_ops_op_lib",
|
|
|
|
|
"@icu//:common",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|