Searched refs:indices3 (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_one_hot_op.py | 42 def construct(self, indices1, indices2, indices3, indices4): argument
44 self.one_hot_3(indices3), self.one_hot_4(indices4))
54 indices3 = Tensor(np.array([[0, 1], [1, 0]]).astype(np.int32))
56 output = one_hot(indices1, indices2, indices3, indices4)
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_one_hot_op.py | 42 def construct(self, indices1, indices2, indices3, indices4): argument
44 self.one_hot_3(indices3), self.one_hot_4(indices4))
51 indices3 = Tensor(np.array([[0, 1], [1, 0]]).astype(nptype))
53 output = one_hot_net(indices1, indices2, indices3, indices4)
|
| /third_party/mesa3d/src/gallium/drivers/r300/ |
| D | r300_render.c | 601 uint16_t indices3[3]; in r300_draw_elements() local
620 memcpy(indices3, ptr + start, 6); in r300_draw_elements()
645 indices3); in r300_draw_elements()
656 info->mode, start, short_count, indices3); in r300_draw_elements()
|
| /third_party/mindspore/tests/st/numpy_native/ |
| D | test_array_ops.py | 1169 indices3 = rand_int(6, 1, 8, 5).astype(onp.int32)
1172 (x, indices1, indices2, indices3, indices4))
|