Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for NodeComparatorID (0.15 sec)

  1. tensorflow/compiler/jit/xla_cluster_util.h

    // the inputs to the node that must be constant.
    extern const char* const kXlaCompileTimeConstantInputsAttr;
    
    using OrderedNodeSet = std::set<Node*, NodeComparatorID>;
    
    // Returns true if `node` has a ref tensor input that it forwards to its output.
    bool HasForwardedRefInput(const Node& node);
    
    // Creates a graph representation to enable cycle detection when clustering.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

              }
    
              // Change b).
              if (HasNodeAttr(copy->def(), kXlaHasHostTransferAttrName)) {
                (*host_graph)->AddControlEdge(copy, sequencer);
              }
            },
            NodeComparatorID());
    
        if (!s.ok()) {
          return s;
        }
      }
      // Reset "_device_ordinal" to placeholder value.
      TF_RETURN_IF_ERROR(ResetDeviceOrdinalToPlaceholderValue(host_graph->get()));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/mark_for_compilation_pass.cc

      // iterator_traits defined and so on.
      std::vector<Node*> sorted_nodes;
      for (Node* node : graph_->op_nodes()) {
        sorted_nodes.push_back(node);
      }
      std::sort(sorted_nodes.begin(), sorted_nodes.end(), NodeComparatorID());
    
      if (*debug_options_.fuel >= std::numeric_limits<int64_t>::max() / 2) {
        // The assumption is that if fuel started out as INT64_MAX, it will forever
        // stay greater than INT64_MAX / 2.
    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