Searched refs:tuple_setitem (Results 1 – 4 of 4) sorted by relevance
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/optimizer/ |
D | opt_test.py | 607 tuple_setitem = Primitive('tuple_setitem') 611 return tuple_setitem(make_tuple(x, y), 0, z) 615 return tuple_setitem(make_tuple(x, y), 1, z) 631 tuple_setitem = Primitive('tuple_setitem') 635 return tuple_getitem(tuple_setitem(t, 0, x), 0) 643 return tuple_getitem(tuple_setitem(t, 0, x), 1)
|
/third_party/mindspore/mindspore/ops/_grad/ |
D | grad_implementations.py | 139 return F.tuple_setitem(C.zeros_like(data), idx, dout), C.zeros_like(idx)
|
/third_party/mindspore/mindspore/ops/ |
D | functional.py | 174 tuple_setitem = Primitive('tuple_setitem') variable
|
/third_party/mindspore/mindspore/ops/operations/ |
D | _grad_ops.py | 1986 self.tuple_setitem = Primitive('tuple_setitem') 2001 dy_shape_all = self.tuple_setitem(dy_shape, 0, dy_shape[0] * 8)
|