Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD5 default_visibility = ["//tensorflow/compiler/tf2xla:internal"],
152 "//tensorflow/compiler/tf2xla:common",
153 "//tensorflow/compiler/tf2xla:xla_compilation_device",
154 "//tensorflow/compiler/tf2xla:xla_compiler",
155 "//tensorflow/compiler/tf2xla:xla_context",
156 "//tensorflow/compiler/tf2xla:xla_helpers",
157 "//tensorflow/compiler/tf2xla:xla_op_registry",
158 "//tensorflow/compiler/tf2xla:xla_resource",
159 "//tensorflow/compiler/tf2xla/lib:broadcast",
160 "//tensorflow/compiler/tf2xla/lib:data_format",
[all …]
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
DBUILD22 "//tensorflow/compiler/tf2xla:common",
23 "//tensorflow/compiler/tf2xla:sharding_util",
24 "//tensorflow/compiler/tf2xla:side_effect_util",
25 "//tensorflow/compiler/tf2xla:xla_compiler",
26 "//tensorflow/compiler/tf2xla:xla_context",
27 "//tensorflow/compiler/tf2xla:xla_helpers",
28 "//tensorflow/compiler/tf2xla:xla_op_registry",
29 "//tensorflow/compiler/tf2xla/kernels:if_op",
30 "//tensorflow/compiler/tf2xla/kernels:while_op",
31 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
[all …]
/external/tensorflow/tensorflow/compiler/jit/
DBUILD29 "//tensorflow/compiler/tf2xla:internal",
41 "//tensorflow/compiler/tf2xla:friends",
84 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
85 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
100 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
101 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
128 "//tensorflow/compiler/tf2xla:xla_compiler",
129 "//tensorflow/compiler/tf2xla:xla_op_registry",
130 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
154 "//tensorflow/compiler/tf2xla:xla_compiler",
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util_test.cc50 tf2xla::Config config; in TEST()
51 tf2xla::Feed* feed = config.add_feed(); in TEST()
58 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
69 tf2xla::Config config; in TEST()
74 tf2xla::Config config; in TEST()
75 tf2xla::Feed* feed = config.add_feed(); in TEST()
81 tf2xla::Config config; in TEST()
87 tf2xla::Config config; in TEST()
88 tf2xla::Feed* feed = config.add_feed(); in TEST()
95 tf2xla::Config config; in TEST()
[all …]
Dxla_jit_compiled_cpu_function_test.cc72 tf2xla::Config SumConfig() { in SumConfig()
73 tf2xla::Config config; in SumConfig()
74 tf2xla::Feed* x = config.add_feed(); in SumConfig()
77 tf2xla::Feed* y = config.add_feed(); in SumConfig()
80 tf2xla::Fetch* sum = config.add_fetch(); in SumConfig()
157 tf2xla::Config SumConfigVariable() { in SumConfigVariable()
165 tf2xla::Config config; in SumConfigVariable()
172 tf2xla::Config config = SumConfig(); in TEST()
234 tf2xla::Config config = SumConfigVariable(); in TEST()
291 tf2xla::Config config = SumConfig(); in TEST()
[all …]
Dgraph_compiler_util.cc55 const protobuf::RepeatedPtrField<tf2xla::Feed>& feeds, in AddArgNodes()
59 const tf2xla::Feed& feed = feeds[arg_index]; in AddArgNodes()
115 const protobuf::RepeatedPtrField<tf2xla::Fetch>& fetches, in AddRetvalNodes()
118 const tf2xla::TensorId& id = fetches[ret_index].id(); in AddRetvalNodes()
149 Graph* graph, const tf2xla::Config& config, in RewriteAndPruneGraph()
166 for (const tf2xla::Feed& feed : config.feed()) { in RewriteAndPruneGraph()
169 for (const tf2xla::Fetch& fetch : config.fetch()) { in RewriteAndPruneGraph()
249 void PopulateXlaArgs(const tf2xla::Config& config, in PopulateXlaArgs()
253 for (const tf2xla::Variable& variable : config.variable()) { in PopulateXlaArgs()
265 Status InitGraph(const GraphDef& graph_def, const tf2xla::Config& config, in InitGraph()
Dmlir_tf2xla.cc58 const tf2xla::Config& config, in ConvertInputInfo()
64 for (const tf2xla::Feed& feed : config.feed()) { in ConvertInputInfo()
87 Status ConvertOutputInfo(const tf2xla::Config& config, in ConvertOutputInfo()
90 for (const tf2xla::Fetch& fetch : config.fetch()) { in ConvertOutputInfo()
100 GraphDef graph_def, const tf2xla::Config& config, in ConvertGraphDefToXlaViaMlir()
Dtf2xla_util.h34 Status ValidateConfig(const tf2xla::Config& config);
42 const tf2xla::Config& config, const OpRegistryInterface* op_registry,
47 Status PruneGraphDefInto(const tf2xla::Config& config, const GraphDef& in,
51 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()
104 tf2xla::Config config = SumConfig(); in TEST()
Dtf2xla.h34 Status ConvertGraphDefToXla(GraphDef graph_def, const tf2xla::Config& config,
45 GraphDef graph_def, const tf2xla::Config& config,
Dgraph_compiler_util.h33 void PopulateXlaArgs(const tf2xla::Config& config,
46 Status InitGraph(const GraphDef& graph_def, const tf2xla::Config& config,
DBUILD31 "//tensorflow/compiler/tf2xla/...",
57 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
76 srcs = ["tf2xla.proto"],
129 name = "tf2xla",
130 srcs = ["tf2xla.cc"],
131 hdrs = ["tf2xla.h"],
141 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
157 hdrs = ["tf2xla.h"],
292 ":tf2xla",
532 "//tensorflow/compiler/tf2xla/lib:util",
[all …]
Dtf2xla.cc58 const tf2xla::Config& config, xla::Client* client, in ConvertGraphToXla()
119 for (const tf2xla::Variable& variable : config.variable()) { in ConvertGraphToXla()
161 Status ConvertGraphDefToXla(GraphDef graph_def, const tf2xla::Config& config, in ConvertGraphDefToXla()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
DBUILD21 "//tensorflow/compiler/tf2xla:xla_compiler",
28 dso = ["//tensorflow/compiler/tf2xla/ops:_xla_ops.so"],
30 "//tensorflow/compiler/tf2xla/ops:xla_ops",
33 "//tensorflow/compiler/tf2xla/ops:gen_xla_ops",
/external/tensorflow/tensorflow/compiler/jit/kernels/
DBUILD5 "//tensorflow/compiler/tf2xla:internal",
23 "//tensorflow/compiler/tf2xla:common",
24 "//tensorflow/compiler/tf2xla:tf2xla_util",
25 "//tensorflow/compiler/tf2xla:xla_compiler",
26 "//tensorflow/compiler/tf2xla:xla_op_registry",
/external/tensorflow/tensorflow/core/tpu/kernels/
DBUILD68 "//tensorflow/compiler/tf2xla:tf2xla_util",
69 "//tensorflow/compiler/tf2xla:xla_compiler",
145 "//tensorflow/compiler/tf2xla:host_compute_metadata_proto",
188 "//tensorflow/compiler/tf2xla:xla_compiler",
281 "//tensorflow/compiler/tf2xla:host_compute_metadata_proto_cc",
300 "//tensorflow/compiler/tf2xla:host_compute_metadata_proto_cc",
301 "//tensorflow/compiler/tf2xla:xla_compiler",
339 "//tensorflow/compiler/tf2xla:host_compute_metadata_proto_cc",
422 "//tensorflow/compiler/tf2xla:xla_compiler",
456 "//tensorflow/compiler/tf2xla:xla_compiler",
[all …]
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
DBUILD89 "//tensorflow/compiler/tf2xla:side_effect_util",
90 "//tensorflow/compiler/tf2xla:tf2xla_util",
135 "//tensorflow/compiler/tf2xla:resource_operation_table",
136 "//tensorflow/compiler/tf2xla:sharding_util",
137 "//tensorflow/compiler/tf2xla:side_effect_util",
138 "//tensorflow/compiler/tf2xla:tf2xla_util",
139 "//tensorflow/compiler/tf2xla:xla_compiler",
214 "//tensorflow/compiler/tf2xla:functionalize_control_flow_util",
215 "//tensorflow/compiler/tf2xla:tf2xla_util",
/external/tensorflow/tensorflow/compiler/aot/
DBUILD50 "//tensorflow/compiler/tf2xla",
51 "//tensorflow/compiler/tf2xla:mlir_tf2xla",
52 "//tensorflow/compiler/tf2xla:tf2xla_proto_cc",
53 "//tensorflow/compiler/tf2xla:tf2xla_util",
54 "//tensorflow/compiler/tf2xla:xla_compiler",
55 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
56 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
135 "//tensorflow/compiler/tf2xla:tf2xla_proto_cc",
136 "//tensorflow/compiler/tf2xla:tf2xla_util",
308 "//tensorflow/compiler/tf2xla:__pkg__",
[all …]
Dcodegen_test.cc192 tf2xla::Config config; in TEST()
193 tf2xla::Feed* feed = config.add_feed(); in TEST()
198 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
201 tf2xla::Variable* variable = config.add_variable(); in TEST()
206 tf2xla::Variable* variable2 = config.add_variable(); in TEST()
210 tf2xla::Variable* variable3 = config.add_variable(); in TEST()
Dcodegen.cc172 Status GenArgMethods(const tf2xla::Config& config, in GenArgMethods()
221 Status GenResultMethods(const tf2xla::Config& config, in GenResultMethods()
231 [](const tf2xla::Variable& var) { return var.readonly(); }); in GenResultMethods()
275 Status GenVariableMethods(const tf2xla::Config& config, in GenVariableMethods()
307 const tf2xla::Variable& var = config.variable(i - config.feed_size()); in GenVariableMethods()
350 Status ValidateFeedFetchCppNames(const tf2xla::Config& config) { in ValidateFeedFetchCppNames()
351 for (const tf2xla::Feed& feed : config.feed()) { in ValidateFeedFetchCppNames()
356 for (const tf2xla::Fetch& fetch : config.fetch()) { in ValidateFeedFetchCppNames()
361 for (const tf2xla::Variable& variable : config.variable()) { in ValidateFeedFetchCppNames()
393 Status GenerateHeader(const CodegenOpts& opts, const tf2xla::Config& config, in GenerateHeader()
[all …]
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py23 from tensorflow.compiler.tf2xla.python import xla as tf2xla unknown
182 tensor = tf2xla.sharding(tensor, sharding=proto.SerializeToString())
184 tensor = tf2xla.sharding(
250 to_tensor = tf2xla.sharding(to_tensor, sharding=sharding)
385 return tf2xla.spmd_full_to_shard_shape(
405 return tf2xla.spmd_shard_to_full_shape(
/external/tensorflow/tensorflow/compiler/xrt/kernels/
DBUILD14 "//tensorflow/compiler/tf2xla:friends",
27 "//tensorflow/compiler/tf2xla:common",
28 "//tensorflow/compiler/tf2xla:xla_compiler",
58 "//tensorflow/compiler/tf2xla:common",
59 "//tensorflow/compiler/tf2xla:xla_compiler",
117 "//tensorflow/compiler/tf2xla:xla_compiler",
/external/tensorflow/tensorflow/compiler/tf2xla/cc/
DBUILD5 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/tf2xla/lib/
DBUILD7 default_visibility = ["//tensorflow/compiler/tf2xla:friends"],
25 "//tensorflow/compiler/tf2xla:common",
43 "//tensorflow/compiler/tf2xla:xla_compiler",
44 "//tensorflow/compiler/tf2xla:xla_helpers",
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-include-tf2xla-fallback.mlir1 // RUN: tf-opt "-xla-legalize-tf=allow-partial-conversion use-tf2xla-fallback=false" -verify-diagno…
2 // RUN: tf-opt "-xla-legalize-tf=use-tf2xla-fallback=true device-type=XLA_CPU_JIT" -verify-diagnost…
3 // RUN: tf-opt "-xla-legalize-tf=allow-partial-conversion use-tf2xla-fallback=true" %s | FileCheck …
4 // RUN: tf-opt "-xla-legalize-tf=allow-partial-conversion use-tf2xla-fallback=true device-type=INVA…

12345