Searched defs:CppFunctionTensorPreHook (Results 1 – 2 of 2) sorted by relevance
11 struct CppFunctionTensorPreHook : public FunctionPreHook { struct15 std::shared_ptr<hooks_list> hooks_;16 size_t value_idx_;
24 CppFunctionTensorPreHook::CppFunctionTensorPreHook( in CppFunctionTensorPreHook() function in torch::autograd::CppFunctionTensorPreHook