Searched refs:CppFunctionSingleTensorPreHook (Results 1 – 3 of 3) sorted by relevance
51 CppFunctionSingleTensorPreHook::CppFunctionSingleTensorPreHook( in CppFunctionSingleTensorPreHook() function in torch::autograd::CppFunctionSingleTensorPreHook56 variable_list CppFunctionSingleTensorPreHook::operator()( in operator ()()
19 struct CppFunctionSingleTensorPreHook : public FunctionPreHook { struct20 CppFunctionSingleTensorPreHook(
567 std::make_unique<CppFunctionSingleTensorPreHook>( in retain_grad()