Searched refs:identity_node1 (Results 1 – 2 of 2) sorted by relevance
59 NodeDef* identity_node1 = graph_def.add_node(); in TestRemoveNodes() local60 identity_node1->set_name("identity_node1"); in TestRemoveNodes()61 identity_node1->set_op("Identity"); in TestRemoveNodes()62 identity_node1->add_input("const_node1"); in TestRemoveNodes()171 NodeDef* identity_node1 = graph_def.add_node(); in TestRemoveChainedNodes() local172 identity_node1->set_name("identity_node1"); in TestRemoveChainedNodes()173 identity_node1->set_op("Identity"); in TestRemoveChainedNodes()174 identity_node1->add_input("const_node1"); in TestRemoveChainedNodes()
374 NodeDef* identity_node1 = in TestMultiPartition() local378 MakeGather("gather1", gather_v2, identity_node1, input_node, &graph_def); in TestMultiPartition()385 {identity_node1, identity_node2, concat_axis_node}, &graph_def); in TestMultiPartition()