| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_util.h | 75 XlaClusterInfo(const string& cluster_name,
78 : cluster_name(cluster_name),
83 const string cluster_name; member in struct:tensorflow::XlaClusterInfo
|
| mark_for_compilation_pass.cc | 784 absl::optional<absl::string_view> cluster_name = GetXlaClusterForNode(*n); local
801 absl::string_view cluster_name = cluster_name_size_pair.first; local
822 absl::optional<absl::string_view> cluster_name; member in struct:tensorflow::EdgeInfo
[all...] |
| partially_decluster_pass.cc | 108 absl::string_view cluster_name = *GetXlaClusterForNode(*n); local
118 if (dst_cluster_name != cluster_name) {
302 absl::string_view cluster_name = *GetXlaClusterForNode(*n); local
307 return !incoming_cluster || *incoming_cluster != cluster_name;
|
| increase_dynamism_for_auto_jit_pass.cc | 238 absl::string_view cluster_name, Node** result) {
246 .WithXlaCluster(string(cluster_name))
306 absl::string_view cluster_name) {
311 g, slice, slice_inputs, cluster_name, &static_shaped_slice));
|
| mark_for_compilation_pass_test.cc | 729 string cluster_name = clusters.begin()->second; local
732 {{"negate", cluster_name}, {"add", cluster_name}});
752 string cluster_name = clusters.begin()->second; local
755 {{"negate", cluster_name},
756 {"identity", cluster_name},
757 {"add", cluster_name}});
[all...] |
| partially_decluster_pass_test.cc | 295 absl::string_view cluster_name) {
297 n->AddAttr(kXlaClusterAttr, string(cluster_name));
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| resolve_cluster.cc | 46 for (const string& cluster_name : cluster_names) {
47 if (StrContains(node_input, cluster_name) &&
48 !StrContains(node.name(), cluster_name)) {
49 new_node->add_input(cluster_name);
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| xla.py | 316 cluster_name = ops.get_default_graph().unique_name('cluster')
317 pivot = control_flow_ops.no_op(name=cluster_name + '/pivot')
318 context = XLACompileContext(name=cluster_name, pivot=pivot)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| tpu.py | [all...] |
