Home
last modified time | relevance | path

Searched defs:out_edge (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc215 for (const Edge* out_edge : n->out_edges()) { in RewriteIntoTempVariable() local
271 for (const Edge* out_edge : n->out_edges()) { in RewriteIntoAddN() local
Dconstant_folding.cc465 for (const Edge* out_edge : added_nodes.first->out_edges()) { in GetConstantGraph() local
543 for (const Edge* out_edge : n->out_edges()) { in ReplaceTensorWithConstant() local
Dimmutable_executor_state.cc325 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dscheduler.cc92 for (const Edge* out_edge : curr->out_edges()) { in ComputeAsap() local
237 for (const Edge* out_edge : event.node->out_edges()) { in ComputeSchedule() local
/external/tensorflow/tensorflow/core/graph/
Dcontrol_flow.cc118 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo() local
Dalgorithm.cc69 for (const Edge* out_edge : n->out_edges()) { in DFSFromHelper() local
79 for (const Edge* out_edge : n->out_edges()) { in DFSFromHelper() local
Dcollective_order.cc54 for (const Edge* out_edge : node->out_edges()) { in DiscoverDataDependencies() local
Dgraph_partition.cc395 for (const Edge* out_edge : node->out_edges()) { in OptimizeControlFlowColocation() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc737 for (const OutEdgeInfo& out_edge : out_edge_info) { in ReplaceNode() local
847 for (const Edge* out_edge : fwd_arg->out_edges()) { in RewriteTensorListWithConstElement() local
884 for (const Edge* out_edge : bwd_arg->out_edges()) { in RewriteTensorListWithConstElement() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc605 for (const SimpleEdge* out_edge : out_edges) { in ContractEdge() local
627 for (const auto& out_edge : dst->out_edges()) { in ContractEdge() local
834 for (const SimpleEdge* out_edge : node->out_edges()) { in SegmentGraph() local
Dsegment_test.cc57 return [node_names](const Edge* out_edge) -> bool { in MakeOutputEdgeCandidateFn()
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass.cc122 for (const Edge* out_edge : n->out_edges()) { in PartiallyDeclusterNode() local
Dnode_matchers.cc348 OutEdge out_edge, in MatchAndExplain()
Dencapsulate_subgraphs_pass.cc1057 for (const Edge* out_edge : orig->out_edges()) { in MakePrunedGraphCopyAndInline() local
Ddeadness_analysis.cc1290 for (const Edge* out_edge : curr_node->out_edges()) { in GetFrameBasedTopologicalOrder() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc267 for (const Edge* out_edge : node->out_edges()) { in FoldConstants() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc318 for (const auto out_edge : loop_switch_node->out_edges()) { in GetOrCreateBeforeEachIterationNode() local
Ddistributed_tpu_rewrite_pass.cc1866 for (const Edge* out_edge : body_node->out_edges()) { in InferXlaShardingFromNeighbors() local
1941 for (const Edge* out_edge : edge->dst()->out_edges()) { in AssignArgsAndRetvalsToCores() local
2956 for (const Edge* out_edge : edge->dst()->out_edges()) { in BuildExecuteNodes() local
3295 for (const Edge* out_edge : e->dst()->out_edges()) { in BuildExecuteNodes() local
Dencapsulate_tpu_computations_pass.cc2054 for (const Edge* out_edge : out_edges) { in RemoveOutputIdentityNodesForWhileV2() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc393 for (const Edge* out_edge : original_input_node->out_edges()) { in TransformGraphToAddAggregatedInputNode() local
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc797 for (const Edge* out_edge : node->out_edges()) { in ClusterizeNodes() local