Searched refs:concat_input_node (Results 1 – 2 of 2) sorted by relevance
148 NodeDef* concat_input_node = in TestSinglePartition() local154 {identity_node, concat_input_node, concat_axis_node}, &graph_def); in TestSinglePartition()160 {identity_node, concat_input_node, in TestSinglePartition()
155 absl::Status GetOffset(const GraphFloat32& graph, NodeId concat_input_node, in GetOffset() argument160 IsMeanNode(graph, graph.FindProducer(concat_input_node), &mean_node)); in GetOffset()