Home
last modified time | relevance | path

Searched defs:IndentedBuffer (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dtriton_combo_kernel.py291 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 Dwrapper.py281 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 Dmemory_planning.py401 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 Dcpp_wrapper_cpu.py629 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 Dcommon.py1511 buffer: IndentedBuffer,
1512 expr: Union[str, CSEVariable, OpsValue, IndentedBuffer], argument
H A Dcpp_utils.py668 buffer: IndentedBuffer,
H A Dcpp.py2312 buffer: Optional[IndentedBuffer] = None,
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dutils.py858 class IndentedBuffer: class
H A Dscheduler.py462 self, buffer: IndentedBuffer, only_once: bool = True