Home
last modified time | relevance | path

Searched refs:castPyCFunctionWithKeywords (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_variable_methods.cpp1250 …{"__add__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_add>), METH_VARAR…
1251 …{"__radd__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_add>), METH_VARA…
1252 …{"__iadd__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_add_>), METH_VAR…
1253 …{"__rmul__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_mul>), METH_VARA…
1254 …{"__mul__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_mul>), METH_VARAR…
1255 …{"__imul__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_mul_>), METH_VAR…
1256 …{"__sub__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_sub>), METH_VARAR…
1257 …{"__isub__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_sub_>), METH_VAR…
1258 …{"__div__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_div>), METH_VARAR…
1259 …{"__truediv__", castPyCFunctionWithKeywords(TypeError_to_NotImplemented_<THPVariable_div>), METH_V…
[all …]
H A Dpython_nn_functions.cpp84 {"_parse_to", castPyCFunctionWithKeywords(THPVariable__parse_to),
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_torch_functions_manual.cpp369 castPyCFunctionWithKeywords(THPVariable_asarray),
373 castPyCFunctionWithKeywords(THPVariable_as_tensor),
378 castPyCFunctionWithKeywords(THPVariable_frombuffer),
382 castPyCFunctionWithKeywords(THPVariable_nonzero),
386 castPyCFunctionWithKeywords(THPVariable_range),
390 castPyCFunctionWithKeywords(THPVariable_sparse_coo_tensor),
394 castPyCFunctionWithKeywords(THPVariable_sparse_compressed_tensor),
398 castPyCFunctionWithKeywords(THPVariable_sparse_csr_tensor),
402 castPyCFunctionWithKeywords(THPVariable_sparse_csc_tensor),
406 castPyCFunctionWithKeywords(THPVariable_sparse_bsr_tensor),
[all …]
H A Dinit.cpp1285 castPyCFunctionWithKeywords(set_grad_enabled),
1296 castPyCFunctionWithKeywords(set_autocast_enabled),
1300 castPyCFunctionWithKeywords(is_autocast_enabled),
1304 castPyCFunctionWithKeywords(set_autocast_dtype),
1308 castPyCFunctionWithKeywords(get_autocast_dtype),
1313 castPyCFunctionWithKeywords(is_autocast_available),
1346 castPyCFunctionWithKeywords(set_anomaly_mode_enabled),
1359 castPyCFunctionWithKeywords(set_multithreading_enabled),
1364 castPyCFunctionWithKeywords(set_view_replay_enabled),
1369 castPyCFunctionWithKeywords(python_exit_dual_level),
[all …]
H A Dpython_nested_functions_manual.cpp33 castPyCFunctionWithKeywords(THPVariable_nested_tensor),
H A Dpython_variable.cpp1750 castPyCFunctionWithKeywords(THPVariable_as_subclass),
1754 castPyCFunctionWithKeywords(THPVariable_make_subclass),
1758 castPyCFunctionWithKeywords(THPVariable_make_wrapper_subclass),
1763 castPyCFunctionWithKeywords(THPVariable_view_func),
1767 castPyCFunctionWithKeywords(THPVariable_view_func_unsafe),
H A Dpython_engine.cpp452 castPyCFunctionWithKeywords(THPEngine_run_backward),
/aosp_15_r20/external/pytorch/torch/csrc/
H A DEvent.cpp261 castPyCFunctionWithKeywords(THPEvent_from_ipc_handle),
265 castPyCFunctionWithKeywords(THPEvent_record),
269 castPyCFunctionWithKeywords(THPEvent_wait),
H A DStorageMethods.cpp637 castPyCFunctionWithKeywords(THPStorage_copy_),
654 castPyCFunctionWithKeywords(THPStorage_fromBuffer),
658 castPyCFunctionWithKeywords(THPStorage_fromFile),
H A DStream.cpp328 castPyCFunctionWithKeywords(THPStream_record_event),
H A DModule.cpp1439 castPyCFunctionWithKeywords(THPModule_setDeterministicAlgorithms),
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpycfunction_helpers.h7 inline PyCFunction castPyCFunctionWithKeywords(PyCFunctionWithKeywords func) { in castPyCFunctionWithKeywords() function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DEvent.cpp191 castPyCFunctionWithKeywords(THCPEvent_from_ipc_handle),
H A DModule.cpp1820 castPyCFunctionWithKeywords(THCPModule_setStream_wrap),
/aosp_15_r20/external/pytorch/torch/csrc/xpu/
H A DModule.cpp417 castPyCFunctionWithKeywords(THXPModule_setStream_wrap),