Searched defs:add_tensor_pre_hook (Results 1 – 2 of 2) sorted by relevance
497 void add_tensor_pre_hook(std::unique_ptr<FunctionPreHook>&& pre_hook) { in add_tensor_pre_hook() function
473 void add_tensor_pre_hook(c10::SafePyObject&& obj, int index) { in add_tensor_pre_hook() function