Searched refs:compute_native_function_declaration (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/dest/ |
| H A D | __init__.py | 8 compute_native_function_declaration as compute_native_function_declaration, unknown
|
| H A D | native_functions.py | 44 def compute_native_function_declaration( function
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_codegen_model.py | 50 dest.compute_native_function_declaration(g, backend_indices[DispatchKey.CPU])
51 dest.compute_native_function_declaration(g, backend_indices[DispatchKey.CUDA])
|
| H A D | test_codegen.py | 360 native_function_decl_gen=dest.compute_native_function_declaration,
370 native_function_decl_gen=dest.compute_native_function_declaration,
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | gen_executorch.py | 332 def compute_native_function_declaration( function
468 native_function_decl_gen=dest.compute_native_function_declaration,
513 native_function_decl_gen=dest.compute_native_function_declaration,
523 native_function_decl_gen=compute_native_function_declaration, # type: ignore[arg-type]
|
| H A D | gen_backend_stubs.py | 378 lambda f: dest.compute_native_function_declaration(
390 else dest.compute_native_function_declaration(
|
| H A D | gen.py | 1436 ] = dest.compute_native_function_declaration,
1493 ] = dest.compute_native_function_declaration,
1928 native_function_decl_gen=dest.compute_native_function_declaration,
2317 "native_declaration": dest.compute_native_function_declaration(
2342 "native_declaration": dest.compute_native_function_declaration(
|