1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
# Description:
# TensorFlow is a computational framework, primarily for use in machine
# learning applications.
package(default_visibility = ["//tensorflow:internal"])
licenses(["notice"]) # Apache 2.0
exports_files(["LICENSE"])
load("/tensorflow/tensorflow", "tf_copts")
load("/tensorflow/tensorflow", "tf_gen_op_wrappers_cc")
cc_library(
name = "cc_op_gen_main",
srcs = [
"ops/cc_op_gen.cc",
"ops/cc_op_gen_main.cc",
],
hdrs = ["ops/cc_op_gen.h"],
copts = tf_copts(),
deps = [
"//tensorflow/core:framework",
],
)
# Generates a library that contains C++ wrappers for ops.
tf_gen_op_wrappers_cc(
name = "cc_ops",
op_lib_names = [
"array_ops",
"attention_ops",
"candidate_sampling_ops",
"control_flow_ops",
"data_flow_ops",
"image_ops",
"io_ops",
"linalg_ops",
"logging_ops",
"math_ops",
"nn_ops",
"no_op",
"parsing_ops",
"random_ops",
"sendrecv_ops",
"sparse_ops",
"state_ops",
"string_ops",
"summary_ops",
"training_ops",
"user_ops",
],
other_hdrs = [
"ops/const_op.h",
"ops/standard_ops.h",
],
other_srcs = [
"ops/const_op.cc",
] + glob(["ops/*_grad.cc"]),
pkg = "//tensorflow/core",
)
cc_binary(
name = "tutorials_example_trainer",
srcs = ["tutorials/example_trainer.cc"],
copts = tf_copts(),
linkopts = [
"-lpthread",
"-lm",
],
deps = [
":cc_ops",
"//tensorflow/core:kernels",
"//tensorflow/core:local",
"//tensorflow/core:tensorflow",
],
)
filegroup(
name = "all_files",
srcs = glob(
["**/*"],
exclude = [
"**/METADATA",
"**/OWNERS",
],
),
visibility = ["//tensorflow:__subpackages__"],
)
|
