Home
last modified time | relevance | path

Searched refs:added_node (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc269 NodeDef* added_node = node_map->GetNode(ctrl_dep_name); in AddControlDependency() local
270 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 …]
Dlayout_optimizer.cc618 NodeDef* added_node = graph_->add_node(); in UpdateAttrValueOfInput() local
619 *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() local
966 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 …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.cc453 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes() local
454 *added_node = old_node; in ReplaceMatchingOpTypes()
458 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes() local
459 *added_node = new_node; in ReplaceMatchingOpTypes()
464 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes() local
465 *added_node = input_node; in ReplaceMatchingOpTypes()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc1653 Status added_node; in InlineFunctionBody() local
1654 Node* clone = g->AddNode(ndef, &added_node); in InlineFunctionBody()
1658 TF_CHECK_OK(added_node); in InlineFunctionBody()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc582 NodeDef* added_node = graph_def.add_node(); in TEST() local
583 *added_node = node_def; in TEST()