Searched refs:PyFunctionTensorPreHook (Results 1 – 5 of 5) sorted by relevance
95 PyFunctionTensorPreHook::PyFunctionTensorPreHook( in PyFunctionTensorPreHook() function in torch::autograd::PyFunctionTensorPreHook103 PyFunctionTensorPreHook::~PyFunctionTensorPreHook() { in ~PyFunctionTensorPreHook()111 auto PyFunctionTensorPreHook::operator()(const variable_list& values) in operator ()()176 void PyFunctionTensorPreHook::compiled_args(CompiledNodeArgs& args) { in compiled_args()
13 struct PyFunctionTensorPreHook : public FunctionPreHook { struct14 PyFunctionTensorPreHook(PyObject* dict, size_t value_idx);15 ~PyFunctionTensorPreHook() override;
86 if (auto pyhook = dynamic_cast<PyFunctionTensorPreHook*>(hook.get())) { in THPCppFunction_traverse()96 dynamic_cast<PyFunctionTensorPreHook*>(pair.second.get())) { in THPCppFunction_traverse()178 fn.add_tensor_pre_hook(std::make_unique<PyFunctionTensorPreHook>( in THPCppFunction_register_hook_dict()
1305 tensor, std::make_unique<PyFunctionTensorPreHook>(obj, 0)); in THPVariable_set_backwards_hooks()2267 dynamic_cast<PyFunctionTensorPreHook*>(hook.get())) { in THPVariable_subclass_traverse()
1404 new PyFunctionTensorPreHook(var->backward_hooks, tensor.output_nr())); in THPFunction__register_hook_dict()