Searched refs:concat_axis_node (Results 1 – 1 of 1) sorted by relevance
146 NodeDef* concat_axis_node = in TestSinglePartition() local154 {identity_node, concat_input_node, concat_axis_node}, &graph_def); in TestSinglePartition()161 concat_input_node_2, concat_axis_node}, in TestSinglePartition()381 NodeDef* concat_axis_node = in TestMultiPartition() local385 {identity_node1, identity_node2, concat_axis_node}, &graph_def); in TestMultiPartition()