Searched refs:enter_node (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 240 Node* enter_node; in CreateEnterNodes() local
252 TF_RETURN_IF_ERROR(builder.Finalize(graph_, &enter_node)); in CreateEnterNodes()
253 enter_nodes_[edge->dst_input()] = enter_node; in CreateEnterNodes()
278 for (Node* enter_node : enter_nodes_) { in CreateMergeNodes()
279 if (enter_node->output_type(0) == DT_RESOURCE) { in CreateMergeNodes()
286 .Input({NodeOut(enter_node, 0), NodeOut(enter_node, 0)}) in CreateMergeNodes()
287 .Device(enter_node->requested_device()) in CreateMergeNodes()
288 .AssignedDevice(enter_node->assigned_device_name()) in CreateMergeNodes()
|
| D | lower_while_op_test.cc | 216 const Node* enter_node = nullptr; in TEST() local
220 enter_node = consumer; in TEST()
228 for (const Node* consumer : enter_node->out_nodes()) { in TEST()
337 const Node* enter_node = nullptr; in TEST() local
341 enter_node = consumer; in TEST()
349 for (const Node* consumer : enter_node->out_nodes()) { in TEST()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 94 const Node* enter_node, const absl::flat_hash_set<Node*> execute_nodes) { in ResourceOnlyUsedForTPUExecuteInLoop() argument
95 for (const Edge* output_edge : enter_node->out_edges()) { in ResourceOnlyUsedForTPUExecuteInLoop()
164 TF_ASSIGN_OR_RETURN(auto enter_node, FindEnterNodeFromTPUExecuteNodeInput( in ExtractExecuteNodeInfo()
166 if (enter_node == nullptr) { in ExtractExecuteNodeInfo()
171 enter_to_execute_nodes[enter_node].insert(edges[j]->dst()); in ExtractExecuteNodeInfo()
606 Node * enter_node, in AddReshardOp()
608 graph->AddEdge(enter_node, 0, unshard_op_node, i); in AddReshardOp()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | collective_order.cc | 40 bool enter_node = in DiscoverDataDependencies() local
42 if (enter_node) { in DiscoverDataDependencies()
57 if (enter_node && s.ok()) { in DiscoverDataDependencies()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 662 NodeDef* enter_node = nullptr; in CheckForDeadFanout() local
667 enter_node = node; in CheckForDeadFanout()
673 if (enter_node != nullptr) { in CheckForDeadFanout()
675 for (const auto& input : enter_node->input()) { in CheckForDeadFanout()
692 << (enter_node ? enter_node->name() : "<n/a>") in CheckForDeadFanout()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 429 Node* enter_node = while_ctx->enter_nodes()[i]; in ProcessWhileLoop() local
430 for (const Edge* e : enter_node->in_edges()) { in ProcessWhileLoop()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 1025 Node* enter_node = options.FinalizeBuilder(&enter_builder); in AddDummyShapedNode() local
1027 node = enter_node; in AddDummyShapedNode()
|