Searched refs:ifm2 (Results 1 – 1 of 1) sorted by relevance
| /linux/drivers/accel/ethosu/ |
| H A D | ethosu_gem.c | 146 struct feat_matrix ifm2; member
237 bool ifm, bool ifm2, bool weight, bool scale) in calc_sizes() argument
261 if (ifm2) { in calc_sizes()
262 len = feat_matrix_length(info, &st->ifm2, st->ifm.depth, in calc_sizes()
265 op, st->ifm2.region, st->ifm2.base[0], len); in calc_sizes()
308 bool ifm, bool ifm2) in calc_sizes_elemwise() argument
326 if (ifm2) { in calc_sizes_elemwise()
327 height = st->ifm2.broadcast & 0x1 ? 0 : st->ofm.height[2]; in calc_sizes_elemwise()
328 width = st->ifm2.broadcast & 0x2 ? 0 : st->ofm.width; in calc_sizes_elemwise()
329 depth = st->ifm2.broadcast & 0x4 ? 0 : st->ofm.depth; in calc_sizes_elemwise()
[all …]
|