Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for AddControlInputs (0.12 sec)

  1. tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc

    }
    
    // Returns the data type of the destination of an edge.
    DataType EdgeType(const Edge* edge) {
      return edge->dst()->input_type(edge->dst_input());
    }
    
    // Adds the control inputs of `node` to `*deps`.
    void AddControlInputs(const Node& node, absl::flat_hash_set<Node*>* deps) {
      for (const Edge* edge : node.in_edges()) {
        if (edge->IsControlEdge()) {
          deps->insert(edge->src());
        }
      }
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.1K bytes
    - Viewed (0)
Back to top