Searched defs:tag_nodes (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/backends/qualcomm/partition/ | ||
| H A D | qnn_partitioner.py | 135 def tag_nodes( member in QnnPartitioner |
| /aosp_15_r20/external/executorch/backends/apple/mps/partition/ | ||
| H A D | mps_partitioner.py | 94 def tag_nodes(self, partitions: List[Partition]) -> None: member in MPSPartitioner |
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ | ||
| H A D | wrap.py | 188 def tag_nodes(self, gmod, is_sac): member in TagActivationCheckpoint |