• Home
  • Raw
  • Download

Lines Matching refs:producer

309         const uint32_t producer_id = value->producer;  in xnn_subgraph_rewrite_for_nchw()
358 const uint32_t producer_id = value->producer; in xnn_subgraph_rewrite_for_nchw()
496 value->producer = XNN_INVALID_NODE_ID; in xnn_subgraph_optimize()
519 assert(subgraph->values[output_id].producer == XNN_INVALID_NODE_ID); in xnn_subgraph_optimize()
520 subgraph->values[output_id].producer = n; in xnn_subgraph_optimize()
544 const uint32_t producer_id = value->producer; in xnn_subgraph_optimize()
556 struct xnn_node* producer = &subgraph->nodes[producer_id]; in xnn_subgraph_optimize() local
557 assert(producer->type != xnn_node_type_invalid); in xnn_subgraph_optimize()
563 switch (producer->type) { in xnn_subgraph_optimize()
576 assert(producer->num_outputs == 1); in xnn_subgraph_optimize()
582 subgraph->values[fused_output_id].producer = producer_id; in xnn_subgraph_optimize()
583 producer->outputs[0] = fused_output_id; in xnn_subgraph_optimize()
585 producer->activation.output_min = in xnn_subgraph_optimize()
586 math_max_f32(producer->activation.output_min, consumer->activation.output_min); in xnn_subgraph_optimize()
587 producer->activation.output_max = in xnn_subgraph_optimize()
588 math_min_f32(producer->activation.output_max, consumer->activation.output_max); in xnn_subgraph_optimize()
598 if (producer->type == xnn_node_type_static_constant_pad) { in xnn_subgraph_optimize()
599 assert(producer->num_inputs == 1); in xnn_subgraph_optimize()
600 assert(producer->num_outputs == 1); in xnn_subgraph_optimize()
602 … (producer->params.static_pad.pre_paddings[0] | producer->params.static_pad.post_paddings[0] | in xnn_subgraph_optimize()
603producer->params.static_pad.pre_paddings[3] | producer->params.static_pad.post_paddings[3]) == 0 && in xnn_subgraph_optimize()
604 producer->params.static_pad.padding_value == 0; in xnn_subgraph_optimize()
611 assert(consumer->inputs[0] == producer->outputs[0]); in xnn_subgraph_optimize()
613 …consumer->params.convolution_2d.input_padding_top += producer->params.static_pad.pre_paddings[1… in xnn_subgraph_optimize()
614 …consumer->params.convolution_2d.input_padding_right += producer->params.static_pad.post_paddings[… in xnn_subgraph_optimize()
615 …consumer->params.convolution_2d.input_padding_bottom += producer->params.static_pad.post_paddings[… in xnn_subgraph_optimize()
616 …consumer->params.convolution_2d.input_padding_left += producer->params.static_pad.pre_paddings[2… in xnn_subgraph_optimize()
618 consumer->inputs[0] = producer->inputs[0]; in xnn_subgraph_optimize()
620 const uint32_t fused_input_id = producer->inputs[0]; in xnn_subgraph_optimize()
626 xnn_node_clear(producer); in xnn_subgraph_optimize()
635 assert(consumer->inputs[0] == producer->outputs[0]); in xnn_subgraph_optimize()
638 producer->params.static_pad.pre_paddings[1]; in xnn_subgraph_optimize()
640 producer->params.static_pad.post_paddings[2]; in xnn_subgraph_optimize()
642 producer->params.static_pad.post_paddings[1]; in xnn_subgraph_optimize()
644 producer->params.static_pad.pre_paddings[2]; in xnn_subgraph_optimize()
646 consumer->inputs[0] = producer->inputs[0]; in xnn_subgraph_optimize()
648 const uint32_t fused_input_id = producer->inputs[0]; in xnn_subgraph_optimize()
654 xnn_node_clear(producer); in xnn_subgraph_optimize()