Searched defs:CppFunctionSingleTensorPreHook (Results 1 – 2 of 2) sorted by relevance
19 struct CppFunctionSingleTensorPreHook : public FunctionPreHook { struct25 std::function<at::TensorBase(const at::TensorBase&)> hook_;26 size_t value_idx_;
51 CppFunctionSingleTensorPreHook::CppFunctionSingleTensorPreHook( in CppFunctionSingleTensorPreHook() function in torch::autograd::CppFunctionSingleTensorPreHook