Home
last modified time | relevance | path

Searched refs:_template_from_string (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_micro_gemm.py111 return KernelTemplate._template_from_string(self.DECLARE_KERNEL).render(options)
270 return KernelTemplate._template_from_string(self.TEMPLATE_ENTRY).render(options)
460 result = KernelTemplate._template_from_string(self.TEMPLATE_KERNEL).render(
463 result += KernelTemplate._template_from_string(self.TEMPLATE_ENTRY).render(
711 result += KernelTemplate._template_from_string(self.TEMPLATE_KERNEL).render(
714 result += KernelTemplate._template_from_string(self.TEMPLATE_ENTRY).render(
H A Dcpp_gemm_template.py1043 return self._template_from_string(GEMM_TEMPLATE).render(**options)
H A Dcommon.py2091 def _template_from_string(source): member in KernelTemplate
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/
H A Dgemm_template.py965 res = self._template_from_string(self._get_template()).render(**options)
967 test_runner_code = self._template_from_string(
1294 epilogue_arguments = self._template_from_string(epilogue_template).render(
1297 arguments = self._template_from_string(argument_template).render(
1552 arguments = self._template_from_string(argument_template).render(
1557 epilogue_arguments = self._template_from_string(epilogue_template).render(
1560 arguments = self._template_from_string(argument_template).render(
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/
H A Dck_universal_gemm_template.py239 return self._template_from_string(template_definition).render(
242 ), self._template_from_string(template_type).render(operation_name=op.name())
300 return self._template_from_string(self.gemm_template).render(
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dselect_algorithm.py598 self.template = self._template_from_string(source)