Home
last modified time | relevance | path

Searched defs:out_edges (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc149 std::vector<const Edge*> out_edges(n->out_edges().begin(), in PropagateShapes() local
Dbuild_xla_ops_pass.cc66 std::vector<const Edge*> out_edges(old_node->out_edges().begin(), in MoveOutgoingEdges() local
Dextract_outside_compilation_pass.cc2161 std::vector<const Edge*> out_edges(n->out_edges().begin(), in CopyOutsideCompilationConstNodes() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc231 std::vector<const Edge*> out_edges; in RearrangeOutputEdges() local
474 std::vector<const Edge*> out_edges; in MaybeRewriteIfNode() local
Dtf2xla_util.cc500 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges) { in SetNodeShardingFromNeighbors()
722 std::vector<const Edge*> out_edges; in ReplaceNode() local
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc238 const EdgeSet& out_edges = node->out_edges(); in AddOutputIdentities() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc88 const std::vector<SimpleEdge*>& out_edges() const { return out_edges_; } in out_edges() function in tensorflow::tensorrt::segment::__anon3618f86b0111::SimpleNode
603 std::vector<const SimpleEdge*> out_edges(dst->out_edges().begin(), in ContractEdge() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc214 std::vector<const Edge*> out_edges(node->out_edges().begin(), in GetEngineInfo() local
/external/tensorflow/tensorflow/core/graph/
Dgraph.h148 const EdgeSet& out_edges() const { return out_edges_; } in out_edges() function
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc548 std::vector<const Edge*> out_edges(old_node->out_edges().begin(), in ReplaceCompilationResultNodeWithIdentity() local
3877 std::vector<const Edge*> out_edges(n->out_edges().begin(), in LowerOutsideCompilationFunctionalNodes() local
Dencapsulate_tpu_computations_pass.cc2052 std::vector<const Edge*> out_edges(identity->out_edges().begin(), in RemoveOutputIdentityNodesForWhileV2() local