Home
last modified time | relevance | path

Searched refs:allow_stack_allocation (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_wrapper_cpu.py1215 self.allow_stack_allocation = False
1344 self.allow_stack_allocation = False
1376 self.allow_stack_allocation = False
1463 self.allow_stack_allocation
1830 self.allow_stack_allocation = False
2131 self.allow_stack_allocation = False
H A Dwrapper.py494 self.allow_stack_allocation: Optional[bool] = None
882 self.allow_stack_allocation = False
994 self.allow_stack_allocation = (
995 self.allow_stack_allocation is not False
996 and config.allow_stack_allocation
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_aot_inductor.py106 "allow_stack_allocation": self.allow_stack_allocation,
140 "allow_stack_allocation": self.allow_stack_allocation,
3453 allow_stack_allocation = False variable in AOTInductorTestABICompatibleCpu
3726 allow_stack_allocation = True variable in AOTInductorTestABICompatibleCpuWithStackAllocation
3745 allow_stack_allocation = True variable in AOTInductorTestABICompatibleCpuWithStackAllocationAndMinimalArrayRefInterface
3764 allow_stack_allocation = False variable in AOTInductorTestABICompatibleCuda
3786 allow_stack_allocation = False variable in AOTInductorTestNonABICompatibleCpu
3840 allow_stack_allocation = False variable in AOTInductorTestNonABICompatibleCuda
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dconfig.py694 allow_stack_allocation: bool = ( variable