Maps to functional.gt or functional.eq.
(self, b)
| 593 | return f"TensorRT LLM Tensor: {self.name=} {self.dtype=} {self.shape=}" |
| 594 | |
| 595 | def __xor__(self, b): |
| 596 | ''' |
| 597 | Maps to functional.gt or functional.eq. |
| 598 | ''' |
| 599 | print(f"self.shape: {self.shape}, b.shape: {b.shape}") |
| 600 | a, b = broadcast_helper(self, b) |
| 601 | print(f"a.shape: {a.shape}, b.shape: {b.shape}") |
| 602 | return op_xor(a, b) |
| 603 | |
| 604 | |
| 605 | def _create_tensor(trt_tensor: trt.ITensor, producer: trt.ILayer) -> Tensor: |
nothing calls this directly
no test coverage detected