Searched refs:ifm2 (Results 1 – 1 of 1) sorted by relevance
| /linux/drivers/accel/ethosu/ |
| H A D | ethosu_gem.c | 147 struct feat_matrix ifm2; member
251 bool ifm, bool ifm2, bool weight, bool scale) in calc_sizes() argument
278 if (ifm2) { in calc_sizes()
279 len = feat_matrix_length(info, &st->ifm2, st->ifm.depth, in calc_sizes()
282 op, st->ifm2.region, st->ifm2.base[0], len); in calc_sizes()
325 bool ifm, bool ifm2) in calc_sizes_elemwise() argument
343 if (ifm2) { in calc_sizes_elemwise()
344 height = st->ifm2.broadcast & 0x1 ? 0 : st->ofm.height[2]; in calc_sizes_elemwise()
345 width = st->ifm2.broadcast & 0x2 ? 0 : st->ofm.width; in calc_sizes_elemwise()
346 depth = st->ifm2.broadcast & 0x4 ? 0 : st->ofm.depth; in calc_sizes_elemwise()
[all …]
|