Searched defs:IndentedBuffer (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
H A D | triton_combo_kernel.py | 291 cls, kernel: "ComboKernel", num: int, code: IndentedBuffer 305 cls, kernel: "ComboKernel", code: IndentedBuffer 376 cls, kernel: "ComboKernel", num: int, code: IndentedBuffer 488 self, code: IndentedBuffer, sub_kernel: TritonKernel, num: int 728 def codegen_blocks(self, code: IndentedBuffer) -> None: 1018 self, code: IndentedBuffer, num_kernel: int, uniquify: List[str] 1045 def call_kernel(self, code: IndentedBuffer, name: str) -> None:
|
H A D | wrapper.py | 281 def codegen(self, code: IndentedBuffer) -> None: 293 def codegen(self, code: IndentedBuffer) -> None: 303 def codegen(self, code: IndentedBuffer) -> None: 346 def codegen(self, code: IndentedBuffer) -> None: 359 def codegen(self, code: IndentedBuffer) -> None: 401 def codegen(self, code: IndentedBuffer) -> None: 424 def codegen(self, code: IndentedBuffer) -> None: 443 def codegen(self, code: IndentedBuffer) -> None: 746 def generate_before_suffix(self, result: IndentedBuffer) -> None: 749 def generate_end(self, result: IndentedBuffer) -> None: [all …]
|
H A D | memory_planning.py | 401 def codegen_create(self, wrapper, code: IndentedBuffer): 430 def codegen_destroy(self, wrapper, code: IndentedBuffer): 565 def codegen(self, code: IndentedBuffer): 595 def codegen(self, code: IndentedBuffer):
|
H A D | cpp_wrapper_cpu.py | 629 def codegen_tensor_dtype_var_decl(self, code: IndentedBuffer, name): 642 def codegen_input_size_var_decl(self, code: IndentedBuffer, name): 651 def codegen_input_stride_var_decl(self, code: IndentedBuffer, name):
|
H A D | common.py | 1511 buffer: IndentedBuffer, 1512 expr: Union[str, CSEVariable, OpsValue, IndentedBuffer], argument
|
H A D | cpp_utils.py | 668 buffer: IndentedBuffer,
|
H A D | cpp.py | 2312 buffer: Optional[IndentedBuffer] = None,
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | utils.py | 858 class IndentedBuffer: class
|
H A D | scheduler.py | 462 self, buffer: IndentedBuffer, only_once: bool = True
|