Home
last modified time | relevance | path

Searched refs:tf2xla (Results 1 – 25 of 77) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD6 default_visibility = ["//tensorflow/compiler/tf2xla:internal"],
127 "//tensorflow/compiler/tf2xla:common",
128 "//tensorflow/compiler/tf2xla:xla_compiler",
129 "//tensorflow/compiler/tf2xla/lib:broadcast",
130 "//tensorflow/compiler/tf2xla/lib:data_format",
131 "//tensorflow/compiler/tf2xla/lib:random",
132 "//tensorflow/compiler/tf2xla/lib:scatter",
133 "//tensorflow/compiler/tf2xla/lib:util",
134 "//tensorflow/compiler/tf2xla/ops:xla_ops",
209 "//tensorflow/compiler/tf2xla:common",
[all …]
/external/tensorflow/tensorflow/compiler/jit/
DBUILD6 "//tensorflow/compiler/tf2xla:internal",
13 "//tensorflow/compiler/tf2xla:friends",
55 "//tensorflow/compiler/tf2xla/kernels:xla_cpu_only_ops",
56 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
57 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
69 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
70 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
86 "//tensorflow/compiler/tf2xla:xla_compiler",
87 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
105 "//tensorflow/compiler/tf2xla:xla_compiler",
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util_test.cc48 tf2xla::Config config; in TEST()
49 tf2xla::Feed* feed = config.add_feed(); in TEST()
56 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
67 tf2xla::Config config; in TEST()
72 tf2xla::Config config; in TEST()
73 tf2xla::Feed* feed = config.add_feed(); in TEST()
79 tf2xla::Config config; in TEST()
85 tf2xla::Config config; in TEST()
86 tf2xla::Feed* feed = config.add_feed(); in TEST()
93 tf2xla::Config config; in TEST()
[all …]
Dxla_jit_compiled_cpu_function_test.cc59 tf2xla::Config SumConfig() { in SumConfig()
60 tf2xla::Config config; in SumConfig()
61 tf2xla::Feed* x = config.add_feed(); in SumConfig()
64 tf2xla::Feed* y = config.add_feed(); in SumConfig()
67 tf2xla::Fetch* sum = config.add_fetch(); in SumConfig()
75 tf2xla::Config config = SumConfig(); in TEST()
140 tf2xla::Config config = SumConfig(); in TEST()
DBUILD11 "//tensorflow/compiler/tf2xla/...",
36 "//tensorflow/compiler/tf2xla/kernels:xla_cpu_only_ops",
37 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
56 srcs = ["tf2xla.proto"],
81 name = "tf2xla",
82 srcs = ["tf2xla.cc"],
83 hdrs = ["tf2xla.h"],
91 "//tensorflow/compiler/tf2xla/kernels:xla_cpu_only_ops",
92 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
152 ":tf2xla",
[all …]
Dtf2xla.cc67 const protobuf::RepeatedPtrField<tf2xla::Feed>& feeds, in AddArgNodes()
70 const tf2xla::Feed& feed = feeds[arg_index]; in AddArgNodes()
123 const protobuf::RepeatedPtrField<tf2xla::Fetch>& fetches, in AddRetvalNodes()
126 const tf2xla::TensorId& id = fetches[ret_index].id(); in AddRetvalNodes()
156 Graph* graph, const tf2xla::Config& config, in RewriteAndPruneGraph()
173 for (const tf2xla::Feed& feed : config.feed()) { in RewriteAndPruneGraph()
176 for (const tf2xla::Fetch& fetch : config.fetch()) { in RewriteAndPruneGraph()
256 const tf2xla::Config& config, xla::Client* client, in ConvertGraphToXla()
271 for (const tf2xla::Variable& variable : config.variable()) { in ConvertGraphToXla()
340 Status InitGraph(const GraphDef& graph_def, const tf2xla::Config& config, in InitGraph()
[all …]
Dtf2xla_util.h33 Status ValidateConfig(const tf2xla::Config& config);
41 const tf2xla::Config& config, const OpRegistryInterface* op_registry,
46 Status PruneGraphDefInto(const tf2xla::Config& config, const GraphDef& in,
50 string TensorIdToString(const tf2xla::TensorId& id);
Dtf2xla_test.cc61 tf2xla::Config SumConfig() { in SumConfig()
62 tf2xla::Config config; in SumConfig()
71 tf2xla::Config config = SumConfig(); in TEST()
Dtf2xla_util.cc46 Status ValidateTensorId(const tf2xla::TensorId& id) { in ValidateTensorId()
312 Status ValidateConfig(const tf2xla::Config& config) { in ValidateConfig()
314 for (const tf2xla::Feed& feed : config.feed()) { in ValidateConfig()
321 for (const tf2xla::Fetch& fetch : config.fetch()) { in ValidateConfig()
333 const tf2xla::Config& config, const OpRegistryInterface* op_registry, in AddPlaceholdersForFeeds()
336 const tf2xla::Feed* feed = nullptr; // point to Feed in <config>. in AddPlaceholdersForFeeds()
345 const tf2xla::Feed* feed = &config.feed(i); in AddPlaceholdersForFeeds()
361 const tf2xla::TensorId& feed_id = info.feed->id(); in AddPlaceholdersForFeeds()
431 Status PruneGraphDefInto(const tf2xla::Config& config, const GraphDef& in, in PruneGraphDefInto()
438 for (const tf2xla::Feed& feed : config.feed()) { in PruneGraphDefInto()
[all …]
Dtf2xla.proto3 package tensorflow.tf2xla;
7 option java_package = "org.tensorflow.tf2xla";
51 // Config represents configuration information for tf2xla conversion.
Dhost_compute_metadata.proto3 package tensorflow.tf2xla;
7 option java_package = "org.tensorflow.tf2xla";
Dxla_compiler.h268 tf2xla::HostComputeMetadata host_compute_metadata;
480 std::unordered_map<string, tf2xla::HostTransferMetadata> host_compute_sends_;
481 std::unordered_map<string, tf2xla::HostTransferMetadata> host_compute_recvs_;
/external/tensorflow/tensorflow/compiler/aot/
DBUILD35 "//tensorflow/compiler/tf2xla",
36 "//tensorflow/compiler/tf2xla:cpu_function_runtime",
37 "//tensorflow/compiler/tf2xla:tf2xla_proto",
38 "//tensorflow/compiler/tf2xla:tf2xla_util",
39 "//tensorflow/compiler/tf2xla:xla_compiler",
40 "//tensorflow/compiler/tf2xla/kernels:xla_cpu_only_ops",
41 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
42 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
96 "//tensorflow/compiler/tf2xla:tf2xla_proto",
97 "//tensorflow/compiler/tf2xla:tf2xla_util",
[all …]
Dcodegen.cc167 Status GenArgMethods(const tf2xla::Config& config, in GenArgMethods()
208 Status GenResultMethods(const tf2xla::Config& config, in GenResultMethods()
251 Status GenVariableMethods(const tf2xla::Config& config, in GenVariableMethods()
263 const tf2xla::Variable& var = config.variable(i - config.feed_size()); in GenVariableMethods()
288 const tf2xla::Variable& var = config.variable(i - config.fetch_size()); in GenVariableMethods()
330 Status ValidateFeedFetchCppNames(const tf2xla::Config& config) { in ValidateFeedFetchCppNames()
331 for (const tf2xla::Feed& feed : config.feed()) { in ValidateFeedFetchCppNames()
336 for (const tf2xla::Fetch& fetch : config.fetch()) { in ValidateFeedFetchCppNames()
341 for (const tf2xla::Variable& variable : config.variable()) { in ValidateFeedFetchCppNames()
373 Status GenerateHeader(const CodegenOpts& opts, const tf2xla::Config& config, in GenerateHeader()
[all …]
Dcodegen_test.cc168 tf2xla::Config config; in TEST()
169 tf2xla::Feed* feed = config.add_feed(); in TEST()
174 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
177 tf2xla::Variable* variable = config.add_variable(); in TEST()
181 tf2xla::Variable* variable2 = config.add_variable(); in TEST()
Dtfcompile_main.cc68 tf2xla::Config config; in Main()
76 for (const tf2xla::Fetch& fetch : config.fetch()) { in Main()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
DBUILD27 "//tensorflow/compiler/tf2xla:xla_compiler",
34 dso = ["//tensorflow/compiler/tf2xla/ops:_xla_ops.so"],
36 "//tensorflow/compiler/tf2xla/ops:xla_ops",
39 "//tensorflow/compiler/tf2xla/ops:gen_xla_ops",
/external/tensorflow/tensorflow/compiler/jit/kernels/
DBUILD5 "//tensorflow/compiler/tf2xla:internal",
19 "//tensorflow/compiler/tf2xla:common",
20 "//tensorflow/compiler/tf2xla:tf2xla_util",
21 "//tensorflow/compiler/tf2xla:xla_compiler",
/external/tensorflow/tensorflow/compiler/tf2xla/cc/
DBUILD2 default_visibility = ["//tensorflow/compiler/tf2xla:internal"],
12 deps = ["//tensorflow/compiler/tf2xla/ops:xla_ops"],
23 "//tensorflow/compiler/tf2xla/ops:xla_ops",
/external/tensorflow/tensorflow/compiler/xrt/kernels/
DBUILD14 "//tensorflow/compiler/tf2xla:common",
15 "//tensorflow/compiler/tf2xla:xla_compiler",
43 "//tensorflow/compiler/tf2xla:xla_compiler",
/external/tensorflow/tensorflow/compiler/xrt/tests/
DBUILD27 "//tensorflow/compiler/tf2xla:common",
28 "//tensorflow/compiler/tf2xla:xla_compiler",
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
DBUILD6 default_visibility = ["//tensorflow/compiler/tf2xla:friends"],
39 "//tensorflow/compiler/tf2xla:xla_compiler",
/external/tensorflow/tensorflow/compiler/xrt/
DBUILD29 "//tensorflow/compiler/tf2xla:host_compute_metadata_proto",
58 "//tensorflow/compiler/tf2xla:xla_compiler",
/external/tensorflow/tensorflow/compiler/tests/
DBUILD6 "//tensorflow/compiler/tf2xla:internal",
13 "//tensorflow/compiler/tf2xla:friends",
403 "//tensorflow/compiler/tf2xla/python:xla",
415 "//tensorflow/compiler/tf2xla/python:xla",
750 "//tensorflow/compiler/tf2xla/python:xla",
1001 "//tensorflow/compiler/tf2xla/python:xla",
1057 "//tensorflow/compiler/tf2xla/python:xla",
1145 "//tensorflow/compiler/tf2xla:common",
1255 "//tensorflow/compiler/tf2xla/python:xla",
1271 "//tensorflow/compiler/tf2xla/python:xla",
/external/tensorflow/tensorflow/contrib/resampler/
DBUILD58 "//tensorflow/compiler/tf2xla/kernels:resampler_ops",
116 "//tensorflow/compiler/tf2xla/kernels:resampler_ops",

1234