Searched refs:idx2d (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops_test.py | 383 idx2d = np.random.randint(0, 5, (10, 2))
387 embedded_np2d = embeds[idx2d]
392 idx2d).eval()
|
| /external/mesa3d/src/gallium/drivers/nouveau/codegen/ |
| D | nv50_ir_from_tgsi.cpp | 1686 void adjustTempIndex(int arrayId, int &idx, int &idx2d) const;
2054 Converter::adjustTempIndex(int arrayId, int &idx, int &idx2d) const in adjustTempIndex()
2061 idx2d = 1; in adjustTempIndex()
2083 int idx2d = src.is2D() ? src.getIndex(1) : 0; in fetchSrc() local
2136 adjustTempIndex(arrayid, idx, idx2d); in fetchSrc()
2140 return getArrayForFile(src.getFile(), idx2d)->load( in fetchSrc()
2151 int idx2d = dst.is2D() ? dst.getIndex(1) : 0; in acquireDst() local
2166 adjustTempIndex(arrayid, idx, idx2d); in acquireDst()
2169 return getArrayForFile(f, idx2d)-> acquire(sub.cur->values, idx, c); in acquireDst()
2201 int idx2d = dst.is2D() ? dst.getIndex(1) : 0; in storeDst() local
[all …]
|