Searched refs:identity_node (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | parallel_concat_optimizer.cc | 107 Node* identity_node; in Run() local
108 TF_RETURN_IF_ERROR(identity_def.Finalize(g, &identity_node)); in Run()
113 g->AddControlEdge(identity_node, e->dst()); in Run()
115 g->AddEdge(identity_node, 0, e->dst(), e->dst_input()); in Run()
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 163 NodeDef* identity_node) { in ConvertReadVariableOpToIdentity() argument
164 identity_node->set_name(node.name()); in ConvertReadVariableOpToIdentity()
165 identity_node->set_op("Identity"); in ConvertReadVariableOpToIdentity()
166 (*identity_node->mutable_attr())["T"] = node.attr().at("dtype"); in ConvertReadVariableOpToIdentity()
167 identity_node->add_input(node.input(0)); in ConvertReadVariableOpToIdentity()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | remove_nodes_test.cc | 142 NodeDef* identity_node = graph_def.add_node(); in TestRemoveOutputNodes() local
143 identity_node->set_name("identity_node"); in TestRemoveOutputNodes()
144 identity_node->set_op("Identity"); in TestRemoveOutputNodes()
145 identity_node->add_input("add_node"); in TestRemoveOutputNodes()
|
| D | sparsify_gather_test.cc | 135 NodeDef* identity_node = in TestSinglePartition() local
137 MakeGather("gather", gather_v2, identity_node, input_node, &graph_def); in TestSinglePartition()
154 {identity_node, concat_input_node, concat_axis_node}, &graph_def); in TestSinglePartition()
160 {identity_node, concat_input_node, in TestSinglePartition()
|
| D | transform_utils_test.cc | 444 NodeDef identity_node; in TestReplaceMatchingOpTypes() local
445 identity_node.set_op("Identity"); in TestReplaceMatchingOpTypes()
446 identity_node.set_name(original_name); in TestReplaceMatchingOpTypes()
447 *(identity_node.mutable_input()->Add()) = original_copy.name(); in TestReplaceMatchingOpTypes()
448 new_nodes->push_back(identity_node); in TestReplaceMatchingOpTypes()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils_test.cc | 1373 const NodeDef& identity_node = vectorized->node_def( in TEST() local
1376 EXPECT_EQ(identity_node.input(0), in TEST()
1379 strings::StrCat(identity_node.name(), ":output:0")); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.cc | 999 NodeDef* identity_node = GetNode(identity_name); in GetOrCreateIdentityConsumingSwitch() local
1000 if (identity_node == nullptr) { in GetOrCreateIdentityConsumingSwitch()
1007 identity_node = AddNode(std::move(new_node)); in GetOrCreateIdentityConsumingSwitch()
1009 return identity_node; in GetOrCreateIdentityConsumingSwitch()
|