Lines Matching refs:subgraph
24 struct xnn_subgraph* subgraph = NULL; in xnn_create_subgraph() local
34 subgraph = xnn_allocate_zero_memory(sizeof(struct xnn_subgraph)); in xnn_create_subgraph()
35 if (subgraph == NULL) { in xnn_create_subgraph()
40 subgraph->external_value_ids = external_value_ids; in xnn_create_subgraph()
42 subgraph->values = xnn_allocate_zero_memory(external_value_ids * sizeof(struct xnn_value)); in xnn_create_subgraph()
43 if (subgraph->values == NULL) { in xnn_create_subgraph()
48 subgraph->values[i].id = i; in xnn_create_subgraph()
50 subgraph->num_values = external_value_ids; in xnn_create_subgraph()
51 subgraph->num_reserved_values = external_value_ids; in xnn_create_subgraph()
53 *subgraph_out = subgraph; in xnn_create_subgraph()
57 xnn_delete_subgraph(subgraph); in xnn_create_subgraph()
62 struct xnn_value* xnn_subgraph_new_internal_value(xnn_subgraph_t subgraph) in xnn_subgraph_new_internal_value() argument
64 struct xnn_value* values = subgraph->values; in xnn_subgraph_new_internal_value()
65 const size_t size = subgraph->num_values; in xnn_subgraph_new_internal_value()
66 const size_t capacity = subgraph->num_reserved_values; in xnn_subgraph_new_internal_value()
78 subgraph->num_reserved_values = new_capacity; in xnn_subgraph_new_internal_value()
79 subgraph->values = values; in xnn_subgraph_new_internal_value()
81 subgraph->num_values = size + 1; in xnn_subgraph_new_internal_value()
87 struct xnn_node* xnn_subgraph_new_node(xnn_subgraph_t subgraph) in xnn_subgraph_new_node() argument
89 struct xnn_node* nodes = subgraph->nodes; in xnn_subgraph_new_node()
90 const size_t size = subgraph->num_nodes; in xnn_subgraph_new_node()
91 const size_t capacity = subgraph->num_reserved_nodes; in xnn_subgraph_new_node()
104 subgraph->num_reserved_nodes = new_capacity; in xnn_subgraph_new_node()
105 subgraph->nodes = nodes; in xnn_subgraph_new_node()
107 subgraph->num_nodes = size + 1; in xnn_subgraph_new_node()
114 xnn_subgraph_t subgraph, in xnn_define_convolution_2d() argument
206 if (input_id >= subgraph->num_values) { in xnn_define_convolution_2d()
213 if (filter_id >= subgraph->num_values) { in xnn_define_convolution_2d()
220 if (bias_id >= subgraph->num_values) { in xnn_define_convolution_2d()
227 if (output_id >= subgraph->num_values) { in xnn_define_convolution_2d()
234 struct xnn_node* node = xnn_subgraph_new_node(subgraph); in xnn_define_convolution_2d()
267 xnn_subgraph_t subgraph, in xnn_define_depthwise_convolution_2d() argument
352 if (input_id >= subgraph->num_values) { in xnn_define_depthwise_convolution_2d()
359 if (filter_id >= subgraph->num_values) { in xnn_define_depthwise_convolution_2d()
366 if (bias_id >= subgraph->num_values) { in xnn_define_depthwise_convolution_2d()
373 if (output_id >= subgraph->num_values) { in xnn_define_depthwise_convolution_2d()
380 struct xnn_node* node = xnn_subgraph_new_node(subgraph); in xnn_define_depthwise_convolution_2d()
412 xnn_subgraph_t subgraph, in xnn_define_add2() argument
445 if (input1_id >= subgraph->num_values) { in xnn_define_add2()
452 if (input2_id >= subgraph->num_values) { in xnn_define_add2()
459 if (output_id >= subgraph->num_values) { in xnn_define_add2()
466 struct xnn_node* node = xnn_subgraph_new_node(subgraph); in xnn_define_add2()
485 xnn_subgraph_t subgraph, in xnn_define_multiply2() argument
518 if (input1_id >= subgraph->num_values) { in xnn_define_multiply2()
525 if (input2_id >= subgraph->num_values) { in xnn_define_multiply2()
532 if (output_id >= subgraph->num_values) { in xnn_define_multiply2()
539 struct xnn_node* node = xnn_subgraph_new_node(subgraph); in xnn_define_multiply2()
558 xnn_subgraph_t subgraph, in xnn_define_prelu() argument
569 if (input_id >= subgraph->num_values) { in xnn_define_prelu()
576 if (slope_id >= subgraph->num_values) { in xnn_define_prelu()
583 if (output_id >= subgraph->num_values) { in xnn_define_prelu()
590 struct xnn_node* node = xnn_subgraph_new_node(subgraph); in xnn_define_prelu()
607 xnn_subgraph_t subgraph, in xnn_define_clamp() argument
619 if (input_id >= subgraph->num_values) { in xnn_define_clamp()
626 if (output_id >= subgraph->num_values) { in xnn_define_clamp()
633 struct xnn_node* node = xnn_subgraph_new_node(subgraph); in xnn_define_clamp()
651 xnn_subgraph_t subgraph, in xnn_define_hardswish() argument
661 if (input_id >= subgraph->num_values) { in xnn_define_hardswish()
668 if (output_id >= subgraph->num_values) { in xnn_define_hardswish()
675 struct xnn_node* node = xnn_subgraph_new_node(subgraph); in xnn_define_hardswish()
691 xnn_subgraph_t subgraph, in xnn_define_sigmoid() argument
701 if (input_id >= subgraph->num_values) { in xnn_define_sigmoid()
708 if (output_id >= subgraph->num_values) { in xnn_define_sigmoid()
715 struct xnn_node* node = xnn_subgraph_new_node(subgraph); in xnn_define_sigmoid()
731 xnn_subgraph_t subgraph, in xnn_define_softmax() argument
741 if (input_id >= subgraph->num_values) { in xnn_define_softmax()
748 if (output_id >= subgraph->num_values) { in xnn_define_softmax()
755 struct xnn_node* node = xnn_subgraph_new_node(subgraph); in xnn_define_softmax()
771 xnn_subgraph_t subgraph) in xnn_delete_subgraph() argument
773 if (subgraph != NULL) { in xnn_delete_subgraph()
774 memset(subgraph->nodes, 0, sizeof(struct xnn_node) * subgraph->num_nodes); in xnn_delete_subgraph()
775 xnn_release_memory(subgraph->nodes); in xnn_delete_subgraph()
777 memset(subgraph->values, 0, sizeof(struct xnn_value) * subgraph->num_values); in xnn_delete_subgraph()
778 xnn_release_memory(subgraph->values); in xnn_delete_subgraph()
780 memset(subgraph, 0, sizeof(struct xnn_subgraph)); in xnn_delete_subgraph()
781 xnn_release_memory(subgraph); in xnn_delete_subgraph()