- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _host_compute_call_node (0.13 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
NodeDefBuilder builder(absl::StrCat(xla_cluster_name, "_key_placeholder"), "Placeholder"); builder.Attr("dtype", DT_STRING); builder.Attr("shape", PartialTensorShape({2})); builder.Attr("_host_compute_call_node", xla_cluster_name); Status s = builder.Finalize(&key_def); if (!s.ok()) return s; Node* n = g->AddNode(key_def, &s); if (!s.ok()) return s; return n; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
"Placeholder", opts.op_registry()); TensorShapeProto shape; shape.add_dim()->set_size(2); return opts.WithAttr("shape", shape) .WithAttr("dtype", DT_STRING) .WithAttr("_host_compute_call_node", call_node) .FinalizeBuilder(&node_builder); } Node* RecvAtHost(ops::NodeOut key_input, const string& cluster, const string& new_func_name, const string& oc_cluster,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0)