Searched refs:codegen_alloc_from_pool (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | memory_planning.py | 163 def codegen_alloc_from_pool(self, wrapper): member in Allocation
168 return wrapper.codegen_alloc_from_pool(
417 self.creation_cache[block.codegen_alloc_from_pool(wrapper)] = self.name
575 alloc_from_pool = allocation.codegen_alloc_from_pool(self.wrapper)
|
| H A D | wrapper.py | 1090 def codegen_alloc_from_pool(self, name, offset, dtype, shape, stride) -> str: member in WrapperCodeGen
|
| H A D | cpp_wrapper_cpu.py | 1651 def codegen_alloc_from_pool(self, name, offset, dtype, shape, stride) -> str: member in CppWrapperCpu
|