Searched defs:out_edges (Results 1 – 9 of 9) sorted by relevance
231 std::vector<const Edge*> out_edges; in RearrangeOutputEdges() local468 std::vector<const Edge*> out_edges; in MaybeRewriteIfNode() local
495 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges) { in SetNodeShardingFromNeighbors()716 std::vector<const Edge*> out_edges; in ReplaceNode() local
149 std::vector<const Edge*> out_edges(n->out_edges().begin(), in PropagateShapes() local
66 std::vector<const Edge*> out_edges(old_node->out_edges().begin(), in MoveOutgoingEdges() local
2108 std::vector<const Edge*> out_edges(n->out_edges().begin(), in CopyOutsideCompilationConstNodes() local
81 const std::vector<SimpleEdge*>& out_edges() const { return out_edges_; } in out_edges() function in tensorflow::tensorrt::segment::SimpleNode375 std::vector<const SimpleEdge*> out_edges(dst->out_edges().begin(), in ContractEdge() local
239 const EdgeSet& out_edges = node->out_edges(); in AddOutputIdentities() local
214 std::vector<const Edge*> out_edges(node->out_edges().begin(), in GetEngineInfo() local
140 const EdgeSet& out_edges() const { return out_edges_; } in out_edges() function