generate_workspace_allocation (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/
H A D	rocm_kernel.py	`148 wrapper.generate_workspace_allocation(`
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/
H A D	cuda_kernel.py	`173 wrapper.generate_workspace_allocation(`
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A D	wrapper.py	`1491 def generate_workspace_allocation(self, nbytes, device, zero_fill): member in WrapperCodeGen`
H A D	triton.py	`2821 wrapper.generate_workspace_allocation(`