Searched defs:TensorKey (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/profiler/ |
| H A D | _memory_profiler.py | 103 class TensorKey(Key): class
119 def __lt__(self, other: "TensorKey") -> bool:
402 def __getitem__(self, key: TensorKey):
605 def lookup(self, key: TensorKey) -> int:
610 def bump(self, key: TensorKey) -> None:
615 def delete(self, key: TensorKey) -> None:
637 def set_by_id(self, key: TensorKey, category: Category) -> None:
641 def set_by_key(self, key: TensorKey, category: Category) -> None:
644 def set_by_version(self, key: TensorKey, version: int, category: Category) -> None:
648 self, key: TensorKey, version: int, category: Category
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor_key.h | 27 TensorKey(const Tensor& t) : Tensor(t) {} in TensorKey() function
|