• Home
  • Raw
  • Download

Lines Matching refs:inputs

92 …      values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */,  in xnn_create_runtime_v2()
93 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
94 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
100 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
101 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
113 runtime->opdata[i].shape1.num_dims = values[node->inputs[0]].shape.num_dims; in xnn_create_runtime_v2()
114 runtime->opdata[i].shape2.num_dims = values[node->inputs[1]].shape.num_dims; in xnn_create_runtime_v2()
116 assert(values[node->inputs[0]].layout == xnn_layout_type_nchw); in xnn_create_runtime_v2()
117 assert(values[node->inputs[1]].layout == xnn_layout_type_nchw); in xnn_create_runtime_v2()
118 runtime->opdata[i].shape1.dim[0] = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
119 …runtime->opdata[i].shape1.dim[1] = values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape… in xnn_create_runtime_v2()
120 if (values[node->inputs[0]].shape.num_dims > 2) { in xnn_create_runtime_v2()
121 …mcpy(&runtime->opdata[i].shape1.dim[2], &values[node->inputs[0]].shape.dim[1], (values[node->input… in xnn_create_runtime_v2()
123 runtime->opdata[i].shape2.dim[0] = values[node->inputs[1]].shape.dim[0]; in xnn_create_runtime_v2()
124 …runtime->opdata[i].shape2.dim[1] = values[node->inputs[1]].shape.dim[values[node->inputs[0]].shape… in xnn_create_runtime_v2()
125 if (values[node->inputs[0]].shape.num_dims > 2) { in xnn_create_runtime_v2()
126 …mcpy(&runtime->opdata[i].shape2.dim[2], &values[node->inputs[1]].shape.dim[1], (values[node->input… in xnn_create_runtime_v2()
130 assert(values[node->inputs[0]].layout == xnn_layout_type_nhwc); in xnn_create_runtime_v2()
131 assert(values[node->inputs[1]].layout == xnn_layout_type_nhwc); in xnn_create_runtime_v2()
132 …memcpy(runtime->opdata[i].shape1.dim, values[node->inputs[0]].shape.dim, values[node->inputs[0]].s… in xnn_create_runtime_v2()
133 …memcpy(runtime->opdata[i].shape2.dim, values[node->inputs[1]].shape.dim, values[node->inputs[1]].s… in xnn_create_runtime_v2()
135 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
136 runtime->opdata[i].inputs[1] = node->inputs[1]; in xnn_create_runtime_v2()
147 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
148 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
149 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
155 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
156 runtime->opdata[i].input_height = values[node->inputs[0]].shape.dim[1]; in xnn_create_runtime_v2()
157 runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[2]; in xnn_create_runtime_v2()
158 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
172 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
173 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
174 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
182 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
183 runtime->opdata[i].input_height = values[node->inputs[0]].shape.dim[1]; in xnn_create_runtime_v2()
184 runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[2]; in xnn_create_runtime_v2()
185 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
190 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
191 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
192 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
198 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
199 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
204 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
205 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
206 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
212 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
213 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
217 assert(values[node->inputs[1]].data != NULL); in xnn_create_runtime_v2()
218 assert(values[node->inputs[2]].data != NULL); in xnn_create_runtime_v2()
236 values[node->inputs[1]].data, in xnn_create_runtime_v2()
237 values[node->inputs[2]].data, in xnn_create_runtime_v2()
240 … node->flags | (values[node->inputs[0]].layout == xnn_layout_type_nhwc ? XNN_FLAG_INPUT_NHWC : 0), in xnn_create_runtime_v2()
243 assert(values[node->inputs[0]].layout == xnn_layout_type_nhwc); in xnn_create_runtime_v2()
261 values[node->inputs[1]].data, in xnn_create_runtime_v2()
262 values[node->inputs[2]].data, in xnn_create_runtime_v2()
271 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
272 runtime->opdata[i].input_height = values[node->inputs[0]].shape.dim[1]; in xnn_create_runtime_v2()
273 runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[2]; in xnn_create_runtime_v2()
274 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
279 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
280 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
281 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
289 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
290 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
294 assert(values[node->inputs[1]].data != NULL); in xnn_create_runtime_v2()
295 assert(values[node->inputs[2]].data != NULL); in xnn_create_runtime_v2()
312 values[node->inputs[1]].data, in xnn_create_runtime_v2()
313 values[node->inputs[2]].data, in xnn_create_runtime_v2()
321 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
322 runtime->opdata[i].input_height = values[node->inputs[0]].shape.dim[1]; in xnn_create_runtime_v2()
323 runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[2]; in xnn_create_runtime_v2()
326 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
330 assert(values[node->inputs[1]].data != NULL); in xnn_create_runtime_v2()
331 assert(values[node->inputs[2]].data != NULL); in xnn_create_runtime_v2()
333 assert(values[node->inputs[0]].layout == xnn_layout_type_nchw); in xnn_create_runtime_v2()
350 values[node->inputs[1]].data, in xnn_create_runtime_v2()
351 values[node->inputs[2]].data, in xnn_create_runtime_v2()
357 assert(values[node->inputs[0]].layout == xnn_layout_type_nhwc); in xnn_create_runtime_v2()
375 values[node->inputs[1]].data, in xnn_create_runtime_v2()
376 values[node->inputs[2]].data, in xnn_create_runtime_v2()
385 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
386 runtime->opdata[i].input_height = values[node->inputs[0]].shape.dim[1]; in xnn_create_runtime_v2()
387 runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[2]; in xnn_create_runtime_v2()
389 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
394 if (values[node->inputs[0]].layout == xnn_layout_type_nchw) { in xnn_create_runtime_v2()
398 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
404 assert(values[node->inputs[0]].layout == xnn_layout_type_nhwc); in xnn_create_runtime_v2()
408 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
417 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
418 runtime->opdata[i].input_height = values[node->inputs[0]].shape.dim[1]; in xnn_create_runtime_v2()
419 runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[2]; in xnn_create_runtime_v2()
422 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
434 runtime->opdata[i].shape1.num_dims = values[node->inputs[0]].shape.num_dims; in xnn_create_runtime_v2()
435 runtime->opdata[i].shape2.num_dims = values[node->inputs[1]].shape.num_dims; in xnn_create_runtime_v2()
436 …memcpy(runtime->opdata[i].shape1.dim, values[node->inputs[0]].shape.dim, values[node->inputs[0]].s… in xnn_create_runtime_v2()
437 …memcpy(runtime->opdata[i].shape2.dim, values[node->inputs[1]].shape.dim, values[node->inputs[1]].s… in xnn_create_runtime_v2()
438 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
439 runtime->opdata[i].inputs[1] = node->inputs[1]; in xnn_create_runtime_v2()
444 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
445 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
446 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
453 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
454 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
459 const size_t num_input_elements = product_all_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
460 const size_t output_channels = values[node->inputs[1]].shape.dim[0]; in xnn_create_runtime_v2()
461 const size_t input_channels = values[node->inputs[1]].shape.dim[1]; in xnn_create_runtime_v2()
467 values[node->inputs[1]].data, in xnn_create_runtime_v2()
468 values[node->inputs[2]].data, in xnn_create_runtime_v2()
477 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
483 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
484 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
485 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
491 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
492 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
496 if (values[node->inputs[0]].layout == xnn_layout_type_nchw) { in xnn_create_runtime_v2()
498 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
504 assert(values[node->inputs[0]].layout == xnn_layout_type_nhwc); in xnn_create_runtime_v2()
507 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
508 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
509 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
518 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
519 …runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[1] * values[node->inputs[0]].sh… in xnn_create_runtime_v2()
520 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
525 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
526 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
527 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
533 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
534 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
539 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
540 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
541 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
548 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
549 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
564 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
565 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
566 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
574 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
575 runtime->opdata[i].input_height = values[node->inputs[0]].shape.dim[1]; in xnn_create_runtime_v2()
576 runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[2]; in xnn_create_runtime_v2()
577 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
587 runtime->opdata[i].shape1.num_dims = values[node->inputs[0]].shape.num_dims; in xnn_create_runtime_v2()
588 runtime->opdata[i].shape2.num_dims = values[node->inputs[1]].shape.num_dims; in xnn_create_runtime_v2()
589 …memcpy(runtime->opdata[i].shape1.dim, values[node->inputs[0]].shape.dim, values[node->inputs[0]].s… in xnn_create_runtime_v2()
590 …memcpy(runtime->opdata[i].shape2.dim, values[node->inputs[1]].shape.dim, values[node->inputs[1]].s… in xnn_create_runtime_v2()
591 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
592 runtime->opdata[i].inputs[1] = node->inputs[1]; in xnn_create_runtime_v2()
602 runtime->opdata[i].shape1.num_dims = values[node->inputs[0]].shape.num_dims; in xnn_create_runtime_v2()
603 runtime->opdata[i].shape2.num_dims = values[node->inputs[1]].shape.num_dims; in xnn_create_runtime_v2()
604 …memcpy(runtime->opdata[i].shape1.dim, values[node->inputs[0]].shape.dim, values[node->inputs[0]].s… in xnn_create_runtime_v2()
605 …memcpy(runtime->opdata[i].shape2.dim, values[node->inputs[1]].shape.dim, values[node->inputs[1]].s… in xnn_create_runtime_v2()
606 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
607 runtime->opdata[i].inputs[1] = node->inputs[1]; in xnn_create_runtime_v2()
619 runtime->opdata[i].shape1.num_dims = values[node->inputs[0]].shape.num_dims; in xnn_create_runtime_v2()
620 runtime->opdata[i].shape2.num_dims = values[node->inputs[1]].shape.num_dims; in xnn_create_runtime_v2()
622 assert(values[node->inputs[0]].layout == xnn_layout_type_nchw); in xnn_create_runtime_v2()
623 assert(values[node->inputs[1]].layout == xnn_layout_type_nchw); in xnn_create_runtime_v2()
624 runtime->opdata[i].shape1.dim[0] = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
625 …runtime->opdata[i].shape1.dim[1] = values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape… in xnn_create_runtime_v2()
626 if (values[node->inputs[0]].shape.num_dims > 2) { in xnn_create_runtime_v2()
627 …mcpy(&runtime->opdata[i].shape1.dim[2], &values[node->inputs[0]].shape.dim[1], (values[node->input… in xnn_create_runtime_v2()
629 runtime->opdata[i].shape2.dim[0] = values[node->inputs[1]].shape.dim[0]; in xnn_create_runtime_v2()
630 …runtime->opdata[i].shape2.dim[1] = values[node->inputs[1]].shape.dim[values[node->inputs[0]].shape… in xnn_create_runtime_v2()
631 if (values[node->inputs[0]].shape.num_dims > 2) { in xnn_create_runtime_v2()
632 …mcpy(&runtime->opdata[i].shape2.dim[2], &values[node->inputs[1]].shape.dim[1], (values[node->input… in xnn_create_runtime_v2()
636 assert(values[node->inputs[0]].layout == xnn_layout_type_nhwc); in xnn_create_runtime_v2()
637 assert(values[node->inputs[1]].layout == xnn_layout_type_nhwc); in xnn_create_runtime_v2()
638 …memcpy(runtime->opdata[i].shape1.dim, values[node->inputs[0]].shape.dim, values[node->inputs[0]].s… in xnn_create_runtime_v2()
639 …memcpy(runtime->opdata[i].shape2.dim, values[node->inputs[1]].shape.dim, values[node->inputs[1]].s… in xnn_create_runtime_v2()
641 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
642 runtime->opdata[i].inputs[1] = node->inputs[1]; in xnn_create_runtime_v2()
647 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
648 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
649 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
655 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
656 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
661 … values[node->inputs[1]].shape.dim[values[node->inputs[1]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
662 … values[node->inputs[1]].shape.dim[values[node->inputs[1]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
663 … values[node->inputs[1]].shape.dim[values[node->inputs[1]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
664 values[node->inputs[1]].data /* negative slope */, in xnn_create_runtime_v2()
670 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
671 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
676 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
677 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
678 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
684 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
685 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
690 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
691 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
692 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
698 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
699 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
710 runtime->opdata[i].shape1 = values[node->inputs[0]].shape; in xnn_create_runtime_v2()
713 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
726 runtime->opdata[i].batch_size = product_all_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
727 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
731 if (values[node->inputs[0]].layout == xnn_layout_type_nchw) { in xnn_create_runtime_v2()
733 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
734 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
735 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
739 assert(values[node->inputs[0]].layout == xnn_layout_type_nhwc); in xnn_create_runtime_v2()
742 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
743 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
744 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
751 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
752 runtime->opdata[i].input_height = values[node->inputs[0]].shape.dim[1]; in xnn_create_runtime_v2()
753 runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[2]; in xnn_create_runtime_v2()
756 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
761 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
762 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
763 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
769 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
770 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
775 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
776 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
777 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
783 runtime->opdata[i].batch_size = product_non_channel_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2()
784 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
794 runtime->opdata[i].shape1.num_dims = values[node->inputs[0]].shape.num_dims; in xnn_create_runtime_v2()
795 runtime->opdata[i].shape2.num_dims = values[node->inputs[1]].shape.num_dims; in xnn_create_runtime_v2()
796 …memcpy(runtime->opdata[i].shape1.dim, values[node->inputs[0]].shape.dim, values[node->inputs[0]].s… in xnn_create_runtime_v2()
797 …memcpy(runtime->opdata[i].shape2.dim, values[node->inputs[1]].shape.dim, values[node->inputs[1]].s… in xnn_create_runtime_v2()
798 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
799 runtime->opdata[i].inputs[1] = node->inputs[1]; in xnn_create_runtime_v2()
811 runtime->opdata[i].shape1.num_dims = values[node->inputs[0]].shape.num_dims; in xnn_create_runtime_v2()
812 runtime->opdata[i].shape2.num_dims = values[node->inputs[1]].shape.num_dims; in xnn_create_runtime_v2()
813 …memcpy(runtime->opdata[i].shape1.dim, values[node->inputs[0]].shape.dim, values[node->inputs[0]].s… in xnn_create_runtime_v2()
814 …memcpy(runtime->opdata[i].shape2.dim, values[node->inputs[1]].shape.dim, values[node->inputs[1]].s… in xnn_create_runtime_v2()
815 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
816 runtime->opdata[i].inputs[1] = node->inputs[1]; in xnn_create_runtime_v2()
827 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* channels */, in xnn_create_runtime_v2()
828 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* input stride */, in xnn_create_runtime_v2()
829 … values[node->inputs[0]].shape.dim[values[node->inputs[0]].shape.num_dims - 1] /* output stride */, in xnn_create_runtime_v2()
835 runtime->opdata[i].batch_size = values[node->inputs[0]].shape.dim[0]; in xnn_create_runtime_v2()
836 runtime->opdata[i].input_height = values[node->inputs[0]].shape.dim[1]; in xnn_create_runtime_v2()
837 runtime->opdata[i].input_width = values[node->inputs[0]].shape.dim[2]; in xnn_create_runtime_v2()
838 runtime->opdata[i].inputs[0] = node->inputs[0]; in xnn_create_runtime_v2()
839 runtime->opdata[i].inputs[1] = node->inputs[1]; in xnn_create_runtime_v2()
948 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
953 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
958 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
959 assert(runtime->blobs[opdata->inputs[1]].data != NULL); in xnn_setup_runtime()
967 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
968 runtime->blobs[opdata->inputs[1]].data, in xnn_setup_runtime()
973 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
981 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
987 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
994 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
999 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1004 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1009 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1014 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1019 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1027 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1032 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1039 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1044 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1051 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1056 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1061 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1066 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1071 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1076 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1085 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1090 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1097 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1102 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1109 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1114 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1115 assert(runtime->blobs[opdata->inputs[1]].data != NULL); in xnn_setup_runtime()
1123 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1124 runtime->blobs[opdata->inputs[1]].data, in xnn_setup_runtime()
1129 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1134 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1139 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1144 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1149 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1154 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1159 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1165 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1170 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1176 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1181 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1186 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1191 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1196 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1201 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1208 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1213 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1214 assert(runtime->blobs[opdata->inputs[1]].data != NULL); in xnn_setup_runtime()
1222 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1223 runtime->blobs[opdata->inputs[1]].data, in xnn_setup_runtime()
1228 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1229 assert(runtime->blobs[opdata->inputs[1]].data != NULL); in xnn_setup_runtime()
1237 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1238 runtime->blobs[opdata->inputs[1]].data, in xnn_setup_runtime()
1243 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1244 assert(runtime->blobs[opdata->inputs[1]].data != NULL); in xnn_setup_runtime()
1252 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1253 runtime->blobs[opdata->inputs[1]].data, in xnn_setup_runtime()
1258 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1263 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1268 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1273 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1278 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1287 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1292 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1301 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1306 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1311 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1316 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1321 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1326 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1331 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1336 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1341 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1346 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1347 assert(runtime->blobs[opdata->inputs[1]].data != NULL); in xnn_setup_runtime()
1355 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1356 runtime->blobs[opdata->inputs[1]].data, in xnn_setup_runtime()
1361 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1362 assert(runtime->blobs[opdata->inputs[1]].data != NULL); in xnn_setup_runtime()
1370 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1371 runtime->blobs[opdata->inputs[1]].data, in xnn_setup_runtime()
1376 assert(runtime->blobs[opdata->inputs[0]].data != NULL); in xnn_setup_runtime()
1377 assert(runtime->blobs[opdata->inputs[1]].data != NULL); in xnn_setup_runtime()
1384 runtime->blobs[opdata->inputs[0]].data, in xnn_setup_runtime()
1385 runtime->blobs[opdata->inputs[1]].data, in xnn_setup_runtime()