Searched refs:GradInterpreterPtr (Results 1 – 4 of 4) sorted by relevance
10 struct TORCH_API GradInterpreterPtr { struct11 …explicit GradInterpreterPtr(const Interpreter* base): base_(base) { TORCH_INTERNAL_ASSERT(base->ke… in GradInterpreterPtr() argument
53 Tensor GradInterpreterPtr::lift(const Tensor& tensor) const { in lift()206 void GradInterpreterPtr::processImpl( in processImpl()212 void GradInterpreterPtr::sendToNextInterpreterImpl( in sendToNextInterpreterImpl()
111 return GradInterpreterPtr(this). method; \
574 py::class_<GradInterpreterPtr>(m, "CGradInterpreterPtr") in initFuncTorchBindings()576 .def("key", &GradInterpreterPtr::key) in initFuncTorchBindings()577 .def("level", &GradInterpreterPtr::level) in initFuncTorchBindings()578 .def("lift", &GradInterpreterPtr::lift) in initFuncTorchBindings()579 .def("prevGradMode", &GradInterpreterPtr::prevGradMode); in initFuncTorchBindings()