Lines Matching refs:elem
165 struct rnndelem *elem;
170 push_rnndoff(lua_State *L, struct rnn *rnn, struct rnndelem *elem,
175 rnndoff->elem = elem;
180 struct rnndelem *elem, uint64_t offset);
181 static int l_rnn_etype_reg(lua_State *L, struct rnn *rnn, struct rnndelem *elem,
214 l_rnn_etype(lua_State *L, struct rnn *rnn, struct rnndelem *elem,
219 DBG("elem=%p (%d), offset=%lu", elem, elem->type, offset);
220 switch (elem->type) {
226 if (elem->width == 64)
228 regval <<= elem->typeinfo.shr;
229 ret = pushdecval(L, rnn, regval, &elem->typeinfo);
232 return l_rnn_etype_reg(L, rnn, elem, offset);
234 return l_rnn_etype_array(L, rnn, elem, offset);
237 printf("unhandled type: %d\n", elem->type);
253 struct rnndelem *elem = rnndoff->elem;
256 for (i = 0; i < elem->subelemsnum; i++) {
257 struct rnndelem *subelem = elem->subelems[i];
272 l_rnn_etype_struct(lua_State *L, struct rnn *rnn, struct rnndelem *elem,
275 push_rnndoff(L, rnn, elem, offset);
295 struct rnndelem *elem = rnndoff->elem;
296 uint64_t offset = rnndoff->offset + (elem->stride * idx);
299 rnndoff->elem->subelemsnum);
305 if (elem->subelemsnum == 1) {
306 return l_rnn_etype(L, rnndoff->rnn, elem->subelems[0], offset);
308 return l_rnn_etype_struct(L, rnndoff->rnn, elem, offset);
319 l_rnn_etype_array(lua_State *L, struct rnn *rnn, struct rnndelem *elem,
322 push_rnndoff(L, rnn, elem, offset);
342 struct rnndelem *elem = rnndoff->elem;
343 struct rnntypeinfo *info = &elem->typeinfo;
372 rnndoff->elem->subelemsnum, bf->typeinfo.type, regval);
409 regval <<= rnndoff->elem->typeinfo.shr;
423 l_rnn_etype_reg(lua_State *L, struct rnn *rnn, struct rnndelem *elem,
426 push_rnndoff(L, rnn, elem, offset);
446 struct rnndelem *elem;
448 elem = rnn_regelem(rnn, name);
449 if (!elem)
452 return l_rnn_etype(L, rnn, elem, elem->offset);
684 struct rnndelem *elem;
692 elem = rnn_regoff(rnn, offset);
693 if (!elem)
696 return l_rnn_etype(L, rnn, elem, elem->offset);