Home
last modified time | relevance | path

Searched refs:add_tensor_pre_hook (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/torch/csrc/autograd/
Dpython_hook.cpp181 args.add_tensor_pre_hook( in compiled_args()
Dpython_cpp_function.cpp178 fn.add_tensor_pre_hook(std::make_unique<PyFunctionTensorPreHook>( in THPCppFunction_register_hook_dict()
Dfunction.h497 void add_tensor_pre_hook(std::unique_ptr<FunctionPreHook>&& pre_hook) { in add_tensor_pre_hook() function
Dvariable.cpp266 fn->add_tensor_pre_hook(std::move(hook_ptr)); in create_cpp_hook()
Dpython_function.cpp1414 cdata->add_tensor_pre_hook(std::move(hook)); in THPFunction__register_hook_dict()
/external/pytorch/torch/csrc/dynamo/
Dcompiled_autograd.h473 void add_tensor_pre_hook(c10::SafePyObject&& obj, int index) { in add_tensor_pre_hook() function