Searched refs:host_compute_control_output_ (Results 1 – 2 of 2) sorted by relevance
356 std::unordered_map<string, xla::XlaOp> host_compute_control_output_; variable
1510 const auto iter = host_compute_control_output_.find(host_compute_name); in GetHostComputeControlDependency()1511 if (iter == host_compute_control_output_.end()) { in GetHostComputeControlDependency()1523 if (host_compute_control_output_.find(host_compute_name) != in SetHostComputeControlDependency()1524 host_compute_control_output_.end()) { in SetHostComputeControlDependency()1529 host_compute_control_output_[host_compute_name] = handle; in SetHostComputeControlDependency()