Lines Matching refs:producer_node
312 struct xnn_node* producer_node = &subgraph->nodes[producer_id]; in xnn_subgraph_rewrite_for_nchw() local
313 …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() local
362 …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()
366 if (producer_node->cluster_leader != node->cluster_leader) { in xnn_subgraph_rewrite_for_nchw()
367 …producer_node->cluster_leader = node->cluster_leader = math_max_u32(producer_node->cluster_leader,… in xnn_subgraph_rewrite_for_nchw()