(self, args, kwargs)
| 2806 | "{x}.dense_shape)`").format(x=self._x, func=func_name)) |
| 2807 | |
| 2808 | def handle(self, args, kwargs): |
| 2809 | if args: |
| 2810 | x, args = args[0], args[1:] |
| 2811 | else: |
| 2812 | kwargs = kwargs.copy() |
| 2813 | x = kwargs.pop(self._x, None) |
| 2814 | if isinstance(x, sparse_tensor.SparseTensor): |
| 2815 | return sparse_tensor.SparseTensor( |
| 2816 | indices=x.indices, |
| 2817 | values=self._original_func(x.values, *args, **kwargs), |
| 2818 | dense_shape=x.dense_shape) |
| 2819 | else: |
| 2820 | return self.NOT_SUPPORTED |
| 2821 | |
| 2822 | |
| 2823 | _UNARY_OPS = [ |