Searched refs:w1_shape (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_gathernd_further.py | 48 def __init__(self, w1_shape, indices_shape, strategy1=None, strategy2=None, strategy3=None): argument
51 self.w1 = Parameter(Tensor(np.ones(w1_shape), dtype=ms.float32), "w1")
64 def __init__(self, w1_shape, indices_shape, strategy1=None, strategy2=None, strategy3=None): argument
67 self.w1 = Parameter(Tensor(np.ones(w1_shape), dtype=ms.float32), "w1")
79 def __init__(self, w1_shape, indices_shape, strategy1=None, strategy2=None, strategy3=None): argument
82 self.w1 = Parameter(Tensor(np.ones(w1_shape), dtype=ms.float32), "w1")
113 w1_shape = [8, 16, 32]
118 net = Net(w1_shape, indices_shape, strategy1, strategy2, strategy3)
125 w1_shape = [8, 16, 32]
130 net = Net(w1_shape, indices_shape, strategy1, strategy2, strategy3)
[all …]
|