Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for out_nodes_ (0.15 sec)

  1. 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)
  2. 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)
Back to top