Searched refs:tensor_pre_hooks (Results 1 – 7 of 7) sorted by relevance
82 std::vector<std::pair<int, int>> tensor_pre_hooks; member442 for (auto& i : fn->tensor_pre_hooks()) { in collect_hooks_from()451 collect_size(_node_call.tensor_pre_hooks.size()); in collect_hooks_from()454 for (const auto& h : _node_call.tensor_pre_hooks) { in collect_hooks_from()476 _node_call.tensor_pre_hooks.emplace_back(fn_id, index); in add_tensor_pre_hook()
570 if (!call.tensor_pre_hooks.empty()) { in _compiled_autograd_impl()572 for (const auto& hook : call.tensor_pre_hooks) { in _compiled_autograd_impl()
45 std::vector<std::unique_ptr<FunctionPreHook>>& tensor_pre_hooks() noexcept in tensor_pre_hooks() function
85 for (const auto& hook : fn.tensor_pre_hooks()) { in THPCppFunction_traverse()
523 tensor_pre_hooks() noexcept { in tensor_pre_hooks() function
810 for (const auto& hook : fn.tensor_pre_hooks()) { in call_tensor_pre_hooks()
495 grad_acc->tensor_pre_hooks().clear(); in THPVariable_clear()