Get the size of the input data. Returns: int: The size of the input data.
(self, *args, **kwargs)
| 155 | |
| 156 | @abstractmethod |
| 157 | def _get_input_size(self, *args, **kwargs) -> int: |
| 158 | """ |
| 159 | Get the size of the input data. |
| 160 | |
| 161 | Returns: |
| 162 | int: The size of the input data. |
| 163 | """ |
| 164 | pass |
| 165 | |
| 166 | @abstractmethod |
| 167 | def _forward_with_mem_ratio(self, *args, mem_ratio=0.0, **kwargs) -> Tuple[float, Tuple]: |