- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for AddControlInputs (0.12 sec)
-
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)