Home
last modified time | relevance | path

Searched refs:index_np_3 (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/st/pynative/
Dtest_tensor_setitem.py303 index_np_1, index_np_2, index_np_3, index_np_4 = 0, 30, 60, 2.0
335 net(input_3d_ms, index_np_3, value_np_1)
337 net(input_3d_ms, index_np_3, value_np_2)
357 …index_np_1, index_np_2, index_np_3, index_np_4, index_np_5 = (0,), (1, 2), (1, 1, 0), (3, 4, 5), (…
368 output_3d_ms_1 = net(input_3d_ms, index_np_3, value_np_1)
369 input_3d_np.itemset(index_np_3, value_np_1)
372 output_3d_ms_2 = net(input_3d_ms, index_np_3, value_np_2)
373 input_3d_np.itemset(index_np_3, value_np_2)
Dtest_tensor_getitem.py288 index_np_1, index_np_2, index_np_3, index_np_4 = 0, 1.0, 30, 60
292 output_3d_ms_2 = net(input_3d_ms, index_np_3)
296 assert np.all(output_3d_ms_2.asnumpy() == input_3d_np.item(index_np_3))
302 net(input_1d_ms, index_np_3)
325 index_np_3 = (1, 2, 3)
330 output_3d_ms = net(input_3d_ms, index_np_3)
332 assert np.all(output_3d_ms.asnumpy() == input_3d_np.item(index_np_3))
/third_party/mindspore/tests/ut/python/ops/
Dtest_tensor_getitem.py127 index_np_1, index_np_2, index_np_3, index_np_4 = 0, 1.0, 30, 60 variable
135 … ('3dTensorItemByInt2', {'block': TensorItemByItem(), 'desc_inputs': [input_3d_ms, index_np_3],}),
156 'desc_inputs': [input_1d_ms, index_np_3]
Dtest_tensor_slice.py660 index_np_1, index_np_2, index_np_3, index_np_4 = 0, 30, 60, 2.0 variable
1239 'desc_inputs': [input_3d_ms, index_np_3, value_np_1]
1243 'desc_inputs': [input_3d_ms, index_np_3, value_np_2]