Searched defs:out_edges (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 146 std::vector<const Edge*> out_edges(n->out_edges().begin(), in PropagateShapes() local
|
| D | build_xla_ops_pass.cc | 66 std::vector<const Edge*> out_edges(old_node->out_edges().begin(), in MoveOutgoingEdges() local
|
| D | extract_outside_compilation_pass.cc | 2164 std::vector<const Edge*> out_edges(n->out_edges().begin(), in CopyOutsideCompilationConstNodes() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 231 std::vector<const Edge*> out_edges; in RearrangeOutputEdges() local
474 std::vector<const Edge*> out_edges; in MaybeRewriteIfNode() local
|
| D | tf2xla_util.cc | 671 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges) { in SetNodeShardingFromNeighbors()
892 std::vector<const Edge*> out_edges; in ReplaceNode() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer_inspection_required_ops_utils.cc | 226 const EdgeSet& out_edges = node->out_edges(); in AddOutputIdentities() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 92 const std::vector<SimpleEdge*>& out_edges() const { return out_edges_; } in out_edges() function in tensorflow::tensorrt::segment::__anon591344890111::SimpleNode
608 std::vector<const SimpleEdge*> out_edges(dst->out_edges().begin(), in ContractEdge() local
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | graph_partition.cc | 117 std::vector<const Edge*> out_edges(node->out_edges().begin(), in PrepareSubgraphForFunctionConversion() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 219 std::vector<const Edge*> out_edges(node->out_edges().begin(), in GetEngineInfo() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 155 const EdgeSet& out_edges() const { return out_edges_; } in out_edges() function
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 568 std::vector<const Edge*> out_edges(old_node->out_edges().begin(), in ReplaceCompilationResultNodeWithIdentity() local
4331 std::vector<const Edge*> out_edges(n->out_edges().begin(), in LowerOutsideCompilationFunctionalNodes() local
|
| D | encapsulate_tpu_computations_pass.cc | 2006 std::vector<const Edge*> out_edges(identity->out_edges().begin(), in RemoveOutputIdentityNodesForWhileV2() local
|