Searched refs:CppFunctionTensorPreHook (Results 1 – 3 of 3) sorted by relevance
24 CppFunctionTensorPreHook::CppFunctionTensorPreHook( in CppFunctionTensorPreHook() function in torch::autograd::CppFunctionTensorPreHook29 variable_list CppFunctionTensorPreHook::operator()( in operator ()()
11 struct CppFunctionTensorPreHook : public FunctionPreHook { struct12 CppFunctionTensorPreHook(std::shared_ptr<hooks_list> hooks, size_t value_idx);
259 std::make_unique<CppFunctionTensorPreHook>(list, self.output_nr()); in create_cpp_hook()264 add_hook(self, std::make_unique<CppFunctionTensorPreHook>(list, 0)); in create_cpp_hook()