Searched refs:add_tensor_pre_hook (Results 1 – 6 of 6) sorted by relevance
181 args.add_tensor_pre_hook( in compiled_args()
178 fn.add_tensor_pre_hook(std::make_unique<PyFunctionTensorPreHook>( in THPCppFunction_register_hook_dict()
497 void add_tensor_pre_hook(std::unique_ptr<FunctionPreHook>&& pre_hook) { in add_tensor_pre_hook() function
266 fn->add_tensor_pre_hook(std::move(hook_ptr)); in create_cpp_hook()
1414 cdata->add_tensor_pre_hook(std::move(hook)); in THPFunction__register_hook_dict()
473 void add_tensor_pre_hook(c10::SafePyObject&& obj, int index) { in add_tensor_pre_hook() function