Home
last modified time | relevance | path

Searched refs:native_function (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_codegen.py38 native_function = dataclasses.replace(DEFAULT_NATIVE_FUNCTION, func=schema)
41 native_function,
51 native_function = dataclasses.replace(DEFAULT_NATIVE_FUNCTION, func=schema)
58 functions_by_signature={schema.signature(): [native_function]},
59 functions_by_schema={specification: native_function},
75 native_function = dataclasses.replace(DEFAULT_NATIVE_FUNCTION, func=schema)
78 native_function,
88 native_function = dataclasses.replace(DEFAULT_NATIVE_FUNCTION, func=schema)
105 functions_by_signature={schema.signature(): [native_function]},
106 functions_by_schema={specification: native_function},
[all …]
H A Dtest_executorch_custom_ops.py23 native_function, _ = NativeFunction.from_yaml(
28 return native_function
H A Dtest_selective_build.py295 native_function, _ = NativeFunction.from_yaml(
300 self.assertTrue(selector.is_native_function_selected(native_function))
/aosp_15_r20/external/pytorch/torchgen/
H A Dgen_vmap_plumbing.py135 def gen_vmap_inplace_plumbing(native_function: NativeFunction) -> str | None:
140 schema = native_function.func
179 def gen_vmap_plumbing_no_returns(native_function: NativeFunction) -> str:
180 schema = native_function.func
200 def gen_vmap_plumbing(native_function: NativeFunction) -> str | None:
201 schema = native_function.func
209 return gen_vmap_plumbing_no_returns(native_function)
220 if "inplace_view" in native_function.tags:
224 return gen_vmap_inplace_plumbing(native_function)
H A Dgen_executorch.py303 (native_function, (kernel_key, metadata))
304 for native_function in native_functions
305 for kernel_key, metadata in kernel_index.get_kernels(native_function).items()
377 for native_function in native_functions:
378 ns_grouped_functions[native_function.namespace].append(native_function)
H A Dgen.py1688 for native_function in native_functions:
1689 ns_native_functions[native_function.namespace].append(native_function)
/aosp_15_r20/external/pytorch/torchgen/executorch/api/
H A Dcustom_ops.py107 for native_function in native_functions:
108 ns_grouped_native_functions[native_function.namespace].append(native_function)
/aosp_15_r20/art/runtime/jni/
H A Djni_internal_test.cc1037 void* native_function = reinterpret_cast<void*>(BogusMethod); in TEST_F() local
1047 JNINativeMethod methods[] = { { nullptr, "()V", native_function } }; in TEST_F()
1054 JNINativeMethod methods[] = { { "notify", nullptr, native_function } }; in TEST_F()
1068 JNINativeMethod methods[] = { { "foo", "()V", native_function } }; in TEST_F()
1075 JNINativeMethod methods[] = { { "equals", "(Ljava/lang/Object;)Z", native_function } }; in TEST_F()
1083 JNINativeMethod methods[] = { { "notify", "()V", native_function } }; in TEST_F()