Searched refs:added_node (Results 1 – 5 of 5) sorted by relevance
269 NodeDef* added_node = node_map->GetNode(ctrl_dep_name); in AddControlDependency() local270 if (added_node == nullptr) { in AddControlDependency()271 added_node = graph->add_node(); in AddControlDependency()272 added_node->set_name(ctrl_dep_name); in AddControlDependency()273 added_node->set_op("Identity"); in AddControlDependency()274 added_node->set_device(node->device()); in AddControlDependency()276 (*added_node->mutable_attr())["T"].set_type(output_type); in AddControlDependency()277 *added_node->add_input() = input_name; in AddControlDependency()278 node_map->AddNode(added_node->name(), added_node); in AddControlDependency()279 node_map->AddOutput(node->name(), added_node->name()); in AddControlDependency()[all …]
618 NodeDef* added_node = graph_->add_node(); in UpdateAttrValueOfInput() local619 *added_node = *input_node; in UpdateAttrValueOfInput()622 added_node->set_name(node_name); in UpdateAttrValueOfInput()624 node_map_->AddNode(node_name, added_node); in UpdateAttrValueOfInput()626 return UpdateAttrValue(added_node, permute); in UpdateAttrValueOfInput()965 NodeDef* added_node = graph_->add_node(); in AddNodeDataFormatOp() local966 added_node->set_name(name); in AddNodeDataFormatOp()967 added_node->set_op(op); in AddNodeDataFormatOp()968 node_map_->AddNode(added_node->name(), added_node); in AddNodeDataFormatOp()969 added_node->set_device(node_->device()); in AddNodeDataFormatOp()[all …]
453 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes() local454 *added_node = old_node; in ReplaceMatchingOpTypes()458 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes() local459 *added_node = new_node; in ReplaceMatchingOpTypes()464 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes() local465 *added_node = input_node; in ReplaceMatchingOpTypes()
1653 Status added_node; in InlineFunctionBody() local1654 Node* clone = g->AddNode(ndef, &added_node); in InlineFunctionBody()1658 TF_CHECK_OK(added_node); in InlineFunctionBody()
582 NodeDef* added_node = graph_def.add_node(); in TEST() local583 *added_node = node_def; in TEST()