Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/autograph/converters/
Dbreak_statements.py76 original_node = node
95 anno.copyanno(original_node, new_while_node, anno.Basic.DIRECTIVES)
117 anno.copyanno(original_node, new_while_node, anno.Basic.DIRECTIVES)
122 original_node = node
146 anno.copyanno(original_node, new_for_node, anno.Basic.EXTRA_LOOP_TEST)
147 anno.copyanno(original_node, new_for_node, anno.Basic.DIRECTIVES)
178 anno.copyanno(original_node, new_for_node, anno.Basic.DIRECTIVES)
Dcontinue_statements.py94 original_node=node)
Dreturn_statements.py267 original_node=node)
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc227 Node* original_node = iter->second; in PostprocessDataEdgesBetweenOutsideCompilations() local
238 g->AddControlEdge(original_node, e->dst()); in PostprocessDataEdgesBetweenOutsideCompilations()
246 absl::StrCat(original_node->name(), ":", node_src_output); in PostprocessDataEdgesBetweenOutsideCompilations()
252 g->AddEdge(original_node, node_src_output, replace_node, dst_input); in PostprocessDataEdgesBetweenOutsideCompilations()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc406 for (const NodeDef* original_node : recomputed_source_nodes) { in RecomputeSubgraph() local
407 VLOG(2) << " " << original_node->name(); in RecomputeSubgraph()
408 recomputed_node_names.insert(original_node->name()); in RecomputeSubgraph()
415 for (const NodeDef* original_node : recomputed_source_nodes) { in RecomputeSubgraph() local
418 AddPrefixToNodeName(original_node->name(), kRecomputedNodePrefix)); in RecomputeSubgraph()
419 copied_node->set_op(original_node->op()); in RecomputeSubgraph()
420 *copied_node->mutable_attr() = original_node->attr(); in RecomputeSubgraph()
421 copied_node->set_device(original_node->device()); in RecomputeSubgraph()
422 for (const string& original_input_name : original_node->input()) { in RecomputeSubgraph()
431 strings::StrCat("^", triggers[original_node]->name()); in RecomputeSubgraph()
Dmodel_pruner.cc179 void ForwardInputs(const NodeDef& original_node, in ForwardInputs() argument
185 ForwardInputsInternal(original_node, nodes_to_delete, in ForwardInputs()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py770 original_node = old_name_to_node_map[node_name]
772 if "_tftrt_op_max_batch_size" in original_node.attr:
773 node_max_batch_size = original_node.attr[
775 elif (original_node.op != "Const" and
777 "_output_shapes" in original_node.attr):
778 node_max_batch_size = _DetectStaticBatchSize(original_node)
781 "'{%s}'s max batch size is %s.", node_name, original_node.op,
/external/tensorflow/tensorflow/core/framework/
Dnode_def.proto71 // C, was result of a fusion of 2 nodes A and B, then 'original_node' would