Home
last modified time | relevance | path

Searched refs:code_string (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/cuda/
H A Djiterator.py13 def __init__(self, code_string: str):
39 pattern, code_string, re.DOTALL
56 self, code_string: str, return_by_ref: bool, num_outputs: int, **kwargs
58 self.code_string = code_string
66 parsed_code = _CodeParser(code_string)
89 self.code_string,
98 def _create_jit_fn(code_string: str, **kwargs) -> Callable:
156 return _JittedFunction(code_string, return_by_ref=False, num_outputs=1, **kwargs)
160 code_string: str, num_outputs: int, **kwargs
186 code_string, return_by_ref=True, num_outputs=num_outputs, **kwargs
/aosp_15_r20/external/pytorch/test/
H A Dtest_jiterator.py18 code_string = "template <typename T> T my_fused_kernel(T x, T y, T alpha, T beta) { return alpha * … variable
19 jitted_fn = create_jit_fn(code_string, alpha=1, beta=1)
84 …code_string = "template <typename T> T conditional(T x, T mask, bool is_train) { return is_train ?…
85 jitted_fn = create_jit_fn(code_string, is_train=False)
98 code_string = '''
102 jitted_fn = create_jit_fn(code_string)
123 … code_string = f"template <typename T> T my_kernel({input_string}) {{ return {function_body}; }}"
124 jitted_fn = create_jit_fn(code_string)
143 …code_string = f"template <typename T> void my_kernel(T input, {output_string}) {{ {function_body} …
145 jitted_fn = create_multi_output_jit_fn(code_string, num_outputs)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/pyct/
H A Dparser.py58 def _unfold_continuations(code_string): argument
60 return code_string.replace('\\\n', '')
63 def dedent_block(code_string): argument
66 code_string = _unfold_continuations(code_string)
68 token_gen = tokenize.generate_tokens(io.StringIO(code_string).readline)
93 return code_string
118 for line, new_line in zip(code_string.split('\n'), new_code.split('\n')):
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/
H A Dcodegen.cpp671 std::string code_string; in generateKernel() local
674 code_string = cuda::cuda_compilation_unit_template.format(env); in generateKernel()
677 code_string = cpu::cpu_compilation_unit_template.format(env); in generateKernel()
681 std::cerr << "fusion code:" << code_string << '\n'; in generateKernel()
683 return code_string; in generateKernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A Djiterator.h16 const std::string& code_string,
30 const std::string& code_string, in CompileAndLaunchKernel() argument
H A Djiterator.cu322 const std::string& code_string, in CompileAndLaunchKernel() argument
347 at::native::jitted_gpu_kernel_dynamic(kernel_name, iter, code_string, extra_args, return_by_ref); in CompileAndLaunchKernel()
/aosp_15_r20/external/python/cpython3/Lib/test/
Dtest_dtrace.py131 code_string = f.read()
135 code = compile(source=code_string,
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.cpp366 const std::string code_string = THPUtils_unpackString(code_string_o); in THCPModule_cudaJiteratorCompileAndLaunchKernel() local
398 code_string, in THCPModule_cudaJiteratorCompileAndLaunchKernel()
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in1838 code_string: str,