Searched refs:host_compute_name (Results 1 – 3 of 3) sorted by relevance
406 Status GetHostComputeControlDependency(const string& host_compute_name,408 Status SetHostComputeControlDependency(const string& host_compute_name,
1269 const string& host_compute_name, xla::XlaOp* handle) { in GetHostComputeControlDependency() argument1270 const auto iter = host_compute_control_output_.find(host_compute_name); in GetHostComputeControlDependency()1273 "No registered control handle for host compute Op '", host_compute_name, in GetHostComputeControlDependency()1282 const string& host_compute_name, const xla::XlaOp& handle) { in SetHostComputeControlDependency() argument1283 if (host_compute_control_output_.find(host_compute_name) != in SetHostComputeControlDependency()1287 host_compute_name); in SetHostComputeControlDependency()1289 host_compute_control_output_[host_compute_name] = handle; in SetHostComputeControlDependency()
406 string host_compute_name; member975 .host_compute_name); in AddHostComputes()997 oc_subgraph.host_compute_name = host_compute->name(); in AddHostComputes()1152 if (n->name() == oc_subgraph.host_compute_name) { in AddShapeInferenceInfo()1161 oc_subgraph.host_compute_name); in AddShapeInferenceInfo()