Fix logical slice assign dtype (#5339)

* refine and add test case

* support ellipsis type slice

* refine

* refine

* support slice assign ellipsis type

* refine

* register fn to localtensor

* make sure input's dtype == update's dtype

Co-authored-by: oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>

oneflow/python/nn/modules/slice.py

@@ -116,6 +116,8 @@ class LogicalSliceAssign(Module):
         self.step = step
 
     def forward(self, x, update):
+        if update.dtype != x.dtype:
+            update = update.to(dtype=x.dtype)
         return flow.F.logical_slice_assign(
             x, update, start=self.start, stop=self.stop, step=self.step
         )