Searched refs:input_1d_ms (Results 1 – 4 of 4) sorted by relevance
/third_party/mindspore/tests/st/pynative/ |
D | test_tensor_setitem.py | 247 input_1d_ms = Tensor(input_1d_np, mstype.float32) 254 output_1d_ms_1 = net(input_1d_ms, value_np_1) 255 output_1d_ms_2 = net(input_1d_ms, value_np_2) 298 input_1d_ms = Tensor(input_1d_np, mstype.float32) 306 output_1d_ms_1 = net(input_1d_ms, index_np_1, value_np_1) 307 output_1d_ms_2 = net(input_1d_ms, index_np_1, value_np_2) 327 net(input_1d_ms, index_np_2, value_np_1) 329 net(input_1d_ms, index_np_2, value_np_2) 331 net(input_1d_ms, index_np_4, value_np_1) 333 net(input_1d_ms, index_np_4, value_np_2) [all …]
|
D | test_tensor_getitem.py | 258 input_1d_ms = Tensor(input_1d_np, mstype.float32) 262 output_ms = net(input_1d_ms) 283 input_1d_ms = Tensor(input_1d_np, mstype.float32) 290 output_1d_ms = net(input_1d_ms, index_np_1) 299 net(input_1d_ms, index_np_2) 302 net(input_1d_ms, index_np_3) 319 input_1d_ms = Tensor(input_1d_np, mstype.float32) 329 output_1d_ms = net(input_1d_ms, index_np_1) 335 net(input_1d_ms, index_np_2)
|
/third_party/mindspore/tests/ut/python/ops/ |
D | test_tensor_getitem.py | 124 input_1d_ms = Tensor(input_1d_np, mstype.float32) variable 132 ('TensorItemByNone', {'block': TensorItemByNone(), 'desc_inputs': [input_1d_ms],}), 133 ('1dTensorItemByInt', {'block': TensorItemByItem(), 'desc_inputs': [input_1d_ms, index_np_1],}), 136 …('1dTensorItemByTuple', {'block': TensorItemByItem(), 'desc_inputs': [input_1d_ms, tuple_index_np_… 148 'desc_inputs': [input_1d_ms, index_np_2] 156 'desc_inputs': [input_1d_ms, index_np_3] 164 'desc_inputs': [input_1d_ms, tuple_index_np_2]
|
D | test_tensor_slice.py | 655 input_1d_ms = Tensor(input_1d_np, mstype.float32) variable 963 'desc_inputs': [input_1d_ms, value_np_1] 967 'desc_inputs': [input_1d_ms, value_np_2] 971 'desc_inputs': [input_1d_ms, index_np_1, value_np_1] 975 'desc_inputs': [input_1d_ms, index_np_1, value_np_2] 995 'desc_inputs': [input_1d_ms, tuple_index_np_1, value_np_1] 999 'desc_inputs': [input_1d_ms, tuple_index_np_1, value_np_2] 1223 'desc_inputs': [input_1d_ms, index_np_2, value_np_1] 1227 'desc_inputs': [input_1d_ms, index_np_2, value_np_2] 1231 'desc_inputs': [input_1d_ms, index_np_4, value_np_1] [all …]
|