Searched refs:identity_node2 (Results 1 – 2 of 2) sorted by relevance
64 NodeDef* identity_node2 = graph_def.add_node(); in TestRemoveNodes() local65 identity_node2->set_name("identity_node2"); in TestRemoveNodes()66 identity_node2->set_op("Identity"); in TestRemoveNodes()67 identity_node2->add_input("const_node2"); in TestRemoveNodes()176 NodeDef* identity_node2 = graph_def.add_node(); in TestRemoveChainedNodes() local177 identity_node2->set_name("identity_node2"); in TestRemoveChainedNodes()178 identity_node2->set_op("Identity"); in TestRemoveChainedNodes()179 identity_node2->add_input("identity_node1"); in TestRemoveChainedNodes()
376 NodeDef* identity_node2 = in TestMultiPartition() local379 MakeGather("gather2", gather_v2, identity_node2, input_node, &graph_def); in TestMultiPartition()385 {identity_node1, identity_node2, concat_axis_node}, &graph_def); in TestMultiPartition()