Get a spatial cache.
(self, key=None)
| 379 | self._spatial_cache[scale_key][key] = value |
| 380 | |
| 381 | def get_spatial_cache(self, key=None): |
| 382 | """ |
| 383 | Get a spatial cache. |
| 384 | """ |
| 385 | scale_key = str(self._scale) |
| 386 | cur_scale_cache = self._spatial_cache.get(scale_key, {}) |
| 387 | if key is None: |
| 388 | return cur_scale_cache |
| 389 | return cur_scale_cache.get(key, None) |
| 390 | |
| 391 | |
| 392 | def sparse_batch_broadcast(input: SparseTensor, other: torch.Tensor) -> torch.Tensor: |
no outgoing calls
no test coverage detected