| /external/tensorflow/tensorflow/core/common_runtime/ |
| parallel_concat_optimizer.cc | 107 Node* identity_node; variable
108 TF_RETURN_IF_ERROR(identity_def.Finalize(g, &identity_node));
113 g->AddControlEdge(identity_node, e->dst());
115 g->AddEdge(identity_node, 0, e->dst(), e->dst_input());
|
| /external/tensorflow/tensorflow/cc/tools/ |
| freeze_saved_model.cc | 163 NodeDef* identity_node) {
164 identity_node->set_name(node.name());
165 identity_node->set_op("Identity");
166 (*identity_node->mutable_attr())["T"] = node.attr().at("dtype");
167 identity_node->add_input(node.input(0));
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| remove_nodes_test.cc | 142 NodeDef* identity_node = graph_def.add_node(); local
143 identity_node->set_name("identity_node");
144 identity_node->set_op("Identity");
145 identity_node->add_input("add_node");
150 context.output_names = {"identity_node"};
160 EXPECT_EQ(1, node_lookup.count("identity_node"));
161 EXPECT_EQ("add_node", node_lookup.at("identity_node")->input(0));
199 context.output_names = {"identity_node"};
|
| sparsify_gather_test.cc | 135 NodeDef* identity_node = local
137 MakeGather("gather", gather_v2, identity_node, input_node, &graph_def);
154 {identity_node, concat_input_node, concat_axis_node}, &graph_def);
160 {identity_node, concat_input_node,
|
| transform_utils_test.cc | 444 NodeDef identity_node;
445 identity_node.set_op("Identity");
446 identity_node.set_name(original_name);
447 *(identity_node.mutable_input()->Add()) = original_copy.name();
448 new_nodes->push_back(identity_node);
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| vectorization_utils_test.cc | 1373 const NodeDef& identity_node = vectorized->node_def( local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/ |
| mutable_graph_view.cc | 999 NodeDef* identity_node = GetNode(identity_name); local
[all...] |
