Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dscheduler.py366 def can_inplace(self, read_dep: dependencies.Dep) -> bool: member in BaseSchedulerNode
1004 def can_inplace(self, read_dep: dependencies.Dep) -> bool: member in SchedulerNode
1235 def can_inplace(self, read_dep: dependencies.Dep) -> bool: member in FusedSchedulerNode
1694 can_inplace: bool = False variable in NodeUser
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dreinplace.py457 def can_inplace(node, mutated_arg): function