Searched refs:get_wrapper_codegen_for_device (Results 1 – 4 of 4) sorted by relevance
30 get_wrapper_codegen_for_device,120 get_wrapper_codegen_for_device("extension_device")124 get_wrapper_codegen_for_device("extension_device", True)
32 get_wrapper_codegen_for_device,89 get_wrapper_codegen_for_device("cpu") == ExtensionWrapperCodegen
53 get_wrapper_codegen_for_device,1530 wrapper_code_gen_cls = get_wrapper_codegen_for_device(
183 def get_wrapper_codegen_for_device(device: str, cpp_wrapper: bool = False): function