|
|
|
@ -305,23 +305,19 @@ def _tensor_setitem_with_slice_v1(data, input_slice, value):
|
|
|
|
|
|
|
|
|
|
@setitem.register("Tensor", "Number", "Number")
|
|
|
|
|
def _tensor_setitem_with_int_v1(data, index, value):
|
|
|
|
|
"""Syntax: A[1] = 3"""
|
|
|
|
|
return compile_utils.tensor_setitem_by_number_with_number(data, index, value)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@setitem.register("Tensor", "Number", "Tensor")
|
|
|
|
|
def _tensor_setitem_with_int_v2(data, index, value):
|
|
|
|
|
"""Syntax: A[1] = Tensor"""
|
|
|
|
|
return compile_utils.tensor_setitem_by_number_with_tensor(data, index, value)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@setitem.register("Tensor", "Ellipsis", "Number")
|
|
|
|
|
def _tensor_setitem_with_ellipsis_v1(data, index, value):
|
|
|
|
|
"""Syntax: A[...] = number."""
|
|
|
|
|
return compile_utils.tensor_setitem_by_ellipsis_with_number(data, index, value)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@setitem.register("Tensor", "Ellipsis", "Tensor")
|
|
|
|
|
def _tensor_setitem_with_ellipsis_v2(data, index, value):
|
|
|
|
|
"""Syntax: A[...] = Tensor."""
|
|
|
|
|
return compile_utils.tensor_setitem_by_ellipsis_with_tensor(data, index, value)
|
|
|
|
|