| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer_inspection_required_ops_utils.cc | 191 Node* identity_node = graph->AddNode(identity_def, &status); in AddInputIdentity() local
195 graph->AddEdge(edge->src(), edge->src_output(), identity_node, 0); in AddInputIdentity()
198 TF_RETURN_IF_ERROR(graph->UpdateEdge(identity_node, 0, node, input_idx)); in AddInputIdentity()
214 Node** identity_node) { in AddOutputIdentities() argument
225 *identity_node = graph->AddNode(identity_def, &status); in AddOutputIdentities()
229 graph->AddEdge(node, src_output, *identity_node, 0); in AddOutputIdentities()
252 Node* identity_node; in AddOutputIdentities() local
253 TF_RETURN_IF_ERROR(add_identity(src_output, identity_name, &identity_node)); in AddOutputIdentities()
256 << identity_node->DebugString(); in AddOutputIdentities()
260 TF_RETURN_IF_ERROR(graph->UpdateEdge(identity_node, 0, dst, dst_input)); in AddOutputIdentities()
[all …]
|
| D | parallel_concat_optimizer.cc | 100 Node* identity_node; in Run() local
101 TF_RETURN_IF_ERROR(identity_def.Finalize(g, &identity_node)); in Run()
106 g->AddControlEdge(identity_node, e->dst()); in Run()
108 g->AddEdge(identity_node, 0, e->dst(), e->dst_input()); in Run()
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 167 NodeDef* identity_node) { in ConvertReadVariableOpToIdentity() argument
168 identity_node->set_name(node.name()); in ConvertReadVariableOpToIdentity()
169 identity_node->set_op("Identity"); in ConvertReadVariableOpToIdentity()
170 (*identity_node->mutable_attr())["T"] = node.attr().at("dtype"); in ConvertReadVariableOpToIdentity()
171 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 | slack_test.cc | 143 NodeDef *identity_node = function_utils::AddNode( in TEST() local
150 strings::StrCat(identity_node->name(), ":output:0"); in TEST()
|
| 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 | 1002 NodeDef* identity_node = GetNode(identity_name); in GetOrCreateIdentityConsumingSwitch() local
1003 if (identity_node == nullptr) { in GetOrCreateIdentityConsumingSwitch()
1010 identity_node = AddNode(std::move(new_node)); in GetOrCreateIdentityConsumingSwitch()
1012 return identity_node; in GetOrCreateIdentityConsumingSwitch()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants.py | 839 identity_node = array_ops.identity(
841 pruned_graph = self._func.prune([], [identity_node.name])()[0]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 772 const NodeDef* identity_node = node_map.GetNode("identity"); in TEST_F() local
773 ASSERT_NE(identity_node, nullptr); in TEST_F()
774 ASSERT_EQ(identity_node->input_size(), 1); in TEST_F()
775 EXPECT_EQ(identity_node->input(0), optimized_name); in TEST_F()
|