Searched refs:u_tensor (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_tensor_slice.py | 600 def construct(self, a, b, c, u_tensor): argument
602 a[b] = u_tensor
611 def construct(self, a, b, c, u_tensor): argument
612 a[b][c] = u_tensor
623 def construct(self, a, u_tensor): argument
624 a[a > 8] = u_tensor
627 a[a <= 5] = u_tensor
637 def construct(self, a, u_tensor): argument
638 a[a > 8][a > 5] = u_tensor
677 u_tensor = Tensor([1], dtype=mstype.float32) variable
[all …]
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_tensor_getitem.py | 1067 def construct(self, a, b, c, u_tensor): argument
1069 a[b] = u_tensor
1075 def construct(self, a, b, c, u_tensor): argument
1076 a[b][c] = u_tensor
1086 def construct(self, a, u_tensor): argument
1087 a[a > 8] = u_tensor
1090 a[a <= 5] = u_tensor
1097 def construct(self, a, u_tensor): argument
1098 a[a > 8][a > 5] = u_tensor
1114 u_tensor = Tensor([1], dtype=mstype.float32)
[all …]
|