Home
last modified time | relevance | path

Searched defs:_is_cuda_graph_recorded_tensor (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/_inductor/
Dcudagraph_trees.py704 def _is_cuda_graph_recorded_tensor(self, t: torch.Tensor) -> bool: member in CUDAWarmupNode
1387 def _is_cuda_graph_recorded_tensor(self, t: torch.Tensor) -> bool: member in CUDAGraphNode