Searched defs:out_edges (Results 1 – 11 of 11) sorted by relevance
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
2161 std::vector<const Edge*> out_edges(n->out_edges().begin(), in CopyOutsideCompilationConstNodes() local
231 std::vector<const Edge*> out_edges; in RearrangeOutputEdges() local474 std::vector<const Edge*> out_edges; in MaybeRewriteIfNode() local
500 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges) { in SetNodeShardingFromNeighbors()722 std::vector<const Edge*> out_edges; in ReplaceNode() local
238 const EdgeSet& out_edges = node->out_edges(); in AddOutputIdentities() local
88 const std::vector<SimpleEdge*>& out_edges() const { return out_edges_; } in out_edges() function in tensorflow::tensorrt::segment::__anon3618f86b0111::SimpleNode603 std::vector<const SimpleEdge*> out_edges(dst->out_edges().begin(), in ContractEdge() local
214 std::vector<const Edge*> out_edges(node->out_edges().begin(), in GetEngineInfo() local
148 const EdgeSet& out_edges() const { return out_edges_; } in out_edges() function
548 std::vector<const Edge*> out_edges(old_node->out_edges().begin(), in ReplaceCompilationResultNodeWithIdentity() local3877 std::vector<const Edge*> out_edges(n->out_edges().begin(), in LowerOutsideCompilationFunctionalNodes() local
2052 std::vector<const Edge*> out_edges(identity->out_edges().begin(), in RemoveOutputIdentityNodesForWhileV2() local