Searched refs:producer_node (Results 1 – 2 of 2) sorted by relevance
312 struct xnn_node* producer_node = &subgraph->nodes[producer_id]; in xnn_subgraph_rewrite_for_nchw() local313 …if ((producer_node->layout_flags & (XNN_LAYOUT_FLAG_COMPATIBLE_NHWC2NCHW | XNN_LAYOUT_FLAG_COMPATI… in xnn_subgraph_rewrite_for_nchw()314 (producer_node->layout_flags & XNN_LAYOUT_FLAG_INCOMPATIBLE_CLUSTER) == 0) in xnn_subgraph_rewrite_for_nchw()316 producer_node->layout_flags &= ~XNN_LAYOUT_FLAG_COMPATIBLE_NCHW2NHWC; in xnn_subgraph_rewrite_for_nchw()317 if (producer_node->cluster_leader != node->cluster_leader) { in xnn_subgraph_rewrite_for_nchw()318 …producer_node->cluster_leader = node->cluster_leader = math_max_u32(producer_node->cluster_leader,… in xnn_subgraph_rewrite_for_nchw()361 struct xnn_node* producer_node = &subgraph->nodes[producer_id]; in xnn_subgraph_rewrite_for_nchw() local362 …if ((producer_node->layout_flags & (XNN_LAYOUT_FLAG_COMPATIBLE_NHWC2NCHW | XNN_LAYOUT_FLAG_COMPATI… in xnn_subgraph_rewrite_for_nchw()363 (producer_node->layout_flags & XNN_LAYOUT_FLAG_INCOMPATIBLE_CLUSTER) == 0) in xnn_subgraph_rewrite_for_nchw()365 producer_node->layout_flags &= ~XNN_LAYOUT_FLAG_COMPATIBLE_NCHW2NHWC; in xnn_subgraph_rewrite_for_nchw()[all …]
212 Node* producer_node = graph.NewNode(); in TEST() local225 ASSERT_TRUE(graph.AddConsumer(producer_node->id, graph_input->id).ok()); in TEST()226 ASSERT_TRUE(graph.SetProducer(producer_node->id, value0->id).ok()); in TEST()234 UnorderedElementsAre(simple_node, producer_node, consumer_node)); in TEST()243 UnorderedElementsAre(producer_node, consumer_node)); in TEST()