Home
last modified time | relevance | path

Searched refs:get_hooks (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/autograd/
Dpython_saved_variable_hooks.cpp76 std::unique_ptr<SavedVariableHooks> PyDefaultSavedVariableHooks::get_hooks() { in get_hooks() function in torch::autograd::PyDefaultSavedVariableHooks
77 auto out = at::SavedTensorDefaultHooks::get_hooks(); in get_hooks()
Dpython_saved_variable_hooks.h30 static std::unique_ptr<SavedVariableHooks> get_hooks();
Dpython_engine.cpp120 return PyDefaultSavedVariableHooks::get_hooks(); in get_default_saved_variable_hooks()
/external/pytorch/aten/src/ATen/
DSavedTensorHooks.h39 get_hooks();
DSavedTensorHooks.cpp74 std::optional<std::pair<SafePyObject, SafePyObject>> SavedTensorDefaultHooks::get_hooks() { in get_hooks() function in at::SavedTensorDefaultHooks