- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for host_compute_builder (0.18 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
// Copy all attributes. for (const auto& attr : call_node->attrs()) { host_compute_builder.Attr(attr.first, attr.second); } // Populate tpu_core assignment. const auto iter = host_compute_core.find(original_oc_name); if (iter != host_compute_core.end()) { int core = iter->second; host_compute_builder.Attr("tpu_core", core); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)