Home
last modified time | relevance | path

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

/third_party/mindspore/tests/st/pynative/
Dtest_tensor_setitem.py247 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 …]
Dtest_tensor_getitem.py258 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/
Dtest_tensor_getitem.py124 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]
Dtest_tensor_slice.py655 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 …]