Searched refs:host_compute_control_output_ (Results 1 – 2 of 2) sorted by relevance
368 std::unordered_map<string, xla::XlaOp> host_compute_control_output_; variable
1546 const auto iter = host_compute_control_output_.find(host_compute_name); in GetHostComputeControlDependency()1547 if (iter == host_compute_control_output_.end()) { in GetHostComputeControlDependency()1559 if (host_compute_control_output_.find(host_compute_name) != in SetHostComputeControlDependency()1560 host_compute_control_output_.end()) { in SetHostComputeControlDependency()1565 host_compute_control_output_[host_compute_name] = handle; in SetHostComputeControlDependency()