Home
last modified time | relevance | path

Searched refs:original_node (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/autograph/converters/
Dcontrol_flow.py444 state_ast_tuple, original_node, argument
464 iter_=original_node.iter,
465 iterate=original_node.target,
472 original_node, body_name, loop_body): argument
488 iter_=original_node.iter,
489 iterate=original_node.target,
493 def _create_for_loop_without_state(self, original_node, body_name, loop_body): argument
505 iter_=original_node.iter,
506 iterate=original_node.target,
Dcontinue_statements.py93 original_node=node)
Dreturn_statements.py277 original_node=node)
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc221 Node* original_node = iter->second; in PostprocessDataEdgesBetweenOutsideCompilations() local
232 g->AddControlEdge(original_node, e->dst()); in PostprocessDataEdgesBetweenOutsideCompilations()
240 absl::StrCat(original_node->name(), ":", node_src_output); in PostprocessDataEdgesBetweenOutsideCompilations()
246 g->AddEdge(original_node, node_src_output, replace_node, dst_input); in PostprocessDataEdgesBetweenOutsideCompilations()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc386 for (const NodeDef* original_node : recomputed_source_nodes) { in RecomputeSubgraph() local
387 VLOG(2) << " " << original_node->name(); in RecomputeSubgraph()
388 recomputed_node_names.insert(original_node->name()); in RecomputeSubgraph()
395 for (const NodeDef* original_node : recomputed_source_nodes) { in RecomputeSubgraph() local
398 AddPrefixToNodeName(original_node->name(), kRecomputedNodePrefix)); in RecomputeSubgraph()
399 copied_node->set_op(original_node->op()); in RecomputeSubgraph()
400 *copied_node->mutable_attr() = original_node->attr(); in RecomputeSubgraph()
401 copied_node->set_device(original_node->device()); in RecomputeSubgraph()
402 for (const string& original_input_name : original_node->input()) { in RecomputeSubgraph()
411 strings::StrCat("^", triggers[original_node]->name()); in RecomputeSubgraph()
Dmodel_pruner.cc161 void ForwardInputs(const NodeDef& original_node, in ForwardInputs() argument
167 ForwardInputsInternal(original_node, nodes_to_delete, in ForwardInputs()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc857 auto& original_node = tree_nodes[node_idx]; in PruneTree() local
858 if (original_node.node_case() == TreeNode::NODE_NOT_SET) { in PruneTree()
869 boosted_trees::trees::DecisionTree::GetChildren(original_node); in PruneTree()
/external/tensorflow/tensorflow/core/framework/
Dnode_def.proto69 // C, was result of a fusion of 2 nodes A and B, then 'original_node' would