- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for out_nodes_ (0.15 sec)
-
tensorflow/compiler/aot/tfcompile.bzl
# First run tfcompile to generate the list of out_nodes. # # Here and below, we set CUDA_VISIBLE_DEVICES='' to prevent the code we # launch from using any GPUs which might be present. This is important # because builds may run concurrently with tests, and tests need to be # able to assume that they have control of the full GPU. out_nodes_file = "out_nodes_" + freeze_name native.genrule(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 19:18:08 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
// increasing its live range. // // See b/221997940 for a real-world example of this. if (n->op_def().name() == "Fill" && n->out_nodes().begin() != n->out_nodes().end() && absl::c_all_of(n->out_nodes(), [&](Node* user) { return GetClusterForNode(user) != cluster; })) { declustered_nodes_.insert(n); } } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)