Invert Tensor division for __truediv__ and __rtruediv__

openvinotoolkit · Nov 1, 2024 · 9cf712c · 9cf712c
1 parent 3ba9b06
commit 9cf712c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nncf/tensor/tensor.py b/nncf/tensor/tensor.py
@@ -117,10 +117,10 @@ def __ipow__(self, other: Union[Tensor, float]) -> Tensor:
         return self
 
     def __truediv__(self, other: Union[Tensor, float]) -> Tensor:
-        return _call_function("_binary_op_nowarn", self, other, operator.truediv)
+        return self * _call_function("_binary_op_nowarn", 1.0, other, operator.truediv)
 
     def __rtruediv__(self, other: Union[Tensor, float]) -> Tensor:
-        return _call_function("_binary_reverse_op_nowarn", self, other, operator.truediv)
+        return other * _call_function("_binary_reverse_op_nowarn", self, 1.0, operator.truediv)
 
     def __itruediv__(self, other: Union[Tensor, float]) -> Tensor:
         self._data /= unwrap_tensor_data(other)